SwatGarg commited on
Commit
bdfad53
1 Parent(s): 90f5ee5

Update generator.py

Browse files
Files changed (1) hide show
  1. generator.py +2 -2
generator.py CHANGED
@@ -15,8 +15,8 @@ def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2"):
15
  try:
16
 
17
  llm = HuggingFaceEndpoint(
18
- repo_id=repo_id,
19
- model_kwargs={"max_length": 128, "api_token": secret_token},
20
  )
21
  return llm
22
  except ValidationError as e:
 
15
  try:
16
 
17
  llm = HuggingFaceEndpoint(
18
+ repo_id=hf_repo_id,
19
+ model_kwargs={"temperature": 0.2, "max_length"=128}, huggingfacehub_api_token = secret_token
20
  )
21
  return llm
22
  except ValidationError as e: