SwatGarg commited on
Commit
45578c9
1 Parent(s): 422e884

Update generator.py

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