Final_Assignment_Template

Sleeping

wishwakankanamg commited on May 24

Commit

6dd38ea

1 Parent(s): 5ebf2af

m

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -170,7 +170,7 @@ def build_graph(provider: str = "huggingface"):
         if not hf_token:
             raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
         llm = HuggingFacePipeline.from_model_id(
-            repo_id="HuggingFaceH4/zephyr-7b-beta",
             task="text-generation",
             pipeline_kwargs=dict(
                 max_new_tokens=512,

         if not hf_token:
             raise ValueError("HF_TOKEN environment variable not set. It's required for Hugging Face provider.")
         llm = HuggingFacePipeline.from_model_id(
+            model_id="HuggingFaceH4/zephyr-7b-beta",
             task="text-generation",
             pipeline_kwargs=dict(
                 max_new_tokens=512,