Spaces:

Abdul-Haseeb
/

Dev-Bot

Sleeping

Abdul-Haseeb commited on Jan 9

Commit

ee9347e

verified ·

1 Parent(s): 52c61c2

Update pipelines.py

Files changed (1) hide show

pipelines.py CHANGED Viewed

@@ -48,8 +48,7 @@ def generate_quiz_pipeline():
     quiz_generation_pipeline.add_component(
         "generator",
         HuggingFaceAPIGenerator(api_type=api_type,
-                                api_params={'model': 'meta-llama/Meta-Llama-3-8B-Instruct',
-                                            "task": "conversational"
                                            },
                                 token=Secret.from_token(HUGGINGFACE_API_KEY),
         )

     quiz_generation_pipeline.add_component(
         "generator",
         HuggingFaceAPIGenerator(api_type=api_type,
+                                api_params={'model': 'meta-llama/Llama-3.1-8B-Instruct'
                                            },
                                 token=Secret.from_token(HUGGINGFACE_API_KEY),
         )