Abdul-Haseeb commited on
Commit
52c61c2
·
verified ·
1 Parent(s): d85098b

Update pipelines.py

Browse files
Files changed (1) hide show
  1. pipelines.py +3 -1
pipelines.py CHANGED
@@ -48,7 +48,9 @@ def generate_quiz_pipeline():
48
  quiz_generation_pipeline.add_component(
49
  "generator",
50
  HuggingFaceAPIGenerator(api_type=api_type,
51
- api_params={'model': 'meta-llama/Meta-Llama-3-8B-Instruct'},
 
 
52
  token=Secret.from_token(HUGGINGFACE_API_KEY),
53
  )
54
  )
 
48
  quiz_generation_pipeline.add_component(
49
  "generator",
50
  HuggingFaceAPIGenerator(api_type=api_type,
51
+ api_params={'model': 'meta-llama/Meta-Llama-3-8B-Instruct',
52
+ "task": "conversational"
53
+ },
54
  token=Secret.from_token(HUGGINGFACE_API_KEY),
55
  )
56
  )