Spaces:

peterpeter8585
/

syai4.1

Sleeping

peterpeter8585 commited on Sep 7

Commit

0b6f430

verified ·

1 Parent(s): e52801c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,8 +79,7 @@ from langchain_huggingface.llms import HuggingFacePipeline
 llm = HuggingFacePipeline.from_model_id(
     model_id="peterpeter8585/zephyr",
-    task="text-generation",
-    pipeline_kwargs={"max_new_tokens": 10},
 )
 from langchain.retrievers import WikipediaRetriever as Wiki
 import gradio as gr

 llm = HuggingFacePipeline.from_model_id(
     model_id="peterpeter8585/zephyr",
+    task="text-generation"
 )
 from langchain.retrievers import WikipediaRetriever as Wiki
 import gradio as gr