Fix NVCF backend

deepset-ai · Mar 18, 2024 · 875773f · 875773f
1 parent a5d6bc1
commit 875773f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/_nvcf_backend.py b/integrations/nvidia/src/haystack_integrations/components/embedders/nvidia/_nvcf_backend.py
@@ -6,6 +6,7 @@
 
 from .backend import EmbedderBackend
 
+MAX_INPUT_STRING_LENGTH = 2048
 MAX_INPUTS = 50
 
 

diff --git a/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/_nvcf_backend.py b/integrations/nvidia/src/haystack_integrations/components/generators/nvidia/_nvcf_backend.py
@@ -6,9 +6,6 @@
 
 from .backend import GeneratorBackend
 
-MAX_INPUT_STRING_LENGTH = 2048
-MAX_INPUTS = 50
-
 
 class NvcfBackend(GeneratorBackend):
     def __init__(
Original file line number	Diff line number	Diff line change
Expand Up		@@ -6,6 +6,7 @@

		from .backend import EmbedderBackend

		MAX_INPUT_STRING_LENGTH = 2048
		MAX_INPUTS = 50


Expand Down