Spaces:

aamirhameed
/

xTwin

Sleeping

aamirhameed commited on Jun 1

Commit

6302c50

verified ·

1 Parent(s): c7deae6

Update knowledge_engine.py

Files changed (1) hide show

knowledge_engine.py CHANGED Viewed

@@ -20,7 +20,7 @@ class KnowledgeManager:
     def _initialize_llm(self):
         # Load local text2text model using HuggingFace pipeline (FLAN-T5 small)
-        local_pipe = pipeline("text2text-generation", model="google/flan-t5-small", max_length=256)
         self.llm = HuggingFacePipeline(pipeline=local_pipe)
     def _initialize_embeddings(self):

     def _initialize_llm(self):
         # Load local text2text model using HuggingFace pipeline (FLAN-T5 small)
+        local_pipe = pipeline("text2text-generation", model="google/flan-t5-small", max_length=500)
         self.llm = HuggingFacePipeline(pipeline=local_pipe)
     def _initialize_embeddings(self):