diff --git a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/__init__.py b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/__init__.py
index 18354ea17..3a315843d 100644
--- a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/__init__.py
+++ b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/__init__.py
@@ -2,5 +2,6 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 from .generator import NvidiaGenerator
+from .models import NvidiaGeneratorModel
 
-__all__ = ["NvidiaGenerator"]
+__all__ = ["NvidiaGenerator", "NvidiaGeneratorModel"]
diff --git a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/generator.py b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/generator.py
index cb24c7fe0..46550baab 100644
--- a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/generator.py
+++ b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/generator.py
@@ -19,7 +19,7 @@ class NvidiaGenerator:
 
     Usage example:
     ```python
-    from haystack_integrations.components.generators.nvidia import NvidiaGenerator
+    from haystack_integrations.components.generators.nvidia import NvidiaGenerator, NvidiaGeneratorModel
 
     generator = NvidiaGenerator(
         model=NvidiaGeneratorModel.NV_LLAMA2_RLHF_70B,