Update app.py
Browse files
app.py
CHANGED
@@ -190,7 +190,7 @@ def generate_text (prompt, chatbot, history, vektordatenbank, retriever, top_p=0
|
|
190 |
#llm = HuggingFaceHub(repo_id=repo_id, model_kwargs=model_kwargs)
|
191 |
#llm = HuggingFaceChain(model=MODEL_NAME_HF, model_kwargs={"temperature": 0.5, "max_length": 128})
|
192 |
# Erstelle eine Pipeline mit den gewünschten Parametern
|
193 |
-
pipe = pipeline("text-generation", model=
|
194 |
|
195 |
# Erstelle eine HuggingFacePipeline-Kette
|
196 |
llm = HuggingFacePipeline(pipeline=pipe)
|
|
|
190 |
#llm = HuggingFaceHub(repo_id=repo_id, model_kwargs=model_kwargs)
|
191 |
#llm = HuggingFaceChain(model=MODEL_NAME_HF, model_kwargs={"temperature": 0.5, "max_length": 128})
|
192 |
# Erstelle eine Pipeline mit den gewünschten Parametern
|
193 |
+
pipe = pipeline("text-generation", model=MODEL_NAME_HF , model_kwargs=model_kwargs)
|
194 |
|
195 |
# Erstelle eine HuggingFacePipeline-Kette
|
196 |
llm = HuggingFacePipeline(pipeline=pipe)
|