Deepakraj2006 commited on
Commit
1aeb2a1
·
verified ·
1 Parent(s): 2dc3938

Update worker.py

Browse files
Files changed (1) hide show
  1. worker.py +1 -1
worker.py CHANGED
@@ -30,7 +30,7 @@ def init_llm():
30
  if not hf_token:
31
  raise ValueError("HUGGINGFACEHUB_API_TOKEN is not set in environment variables.")
32
 
33
- model_id = "tiiuae/falcon-7b-instruct"
34
  hf_pipeline = pipeline("text-generation", model=model_id, device=DEVICE)
35
  llm_pipeline = HuggingFacePipeline(pipeline=hf_pipeline)
36
 
 
30
  if not hf_token:
31
  raise ValueError("HUGGINGFACEHUB_API_TOKEN is not set in environment variables.")
32
 
33
+ model_id = "mistralai/Mistral-7B-Instruct-v0.1"
34
  hf_pipeline = pipeline("text-generation", model=model_id, device=DEVICE)
35
  llm_pipeline = HuggingFacePipeline(pipeline=hf_pipeline)
36