Chris4K commited on
Commit
eeb44aa
1 Parent(s): 99a7864

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -14,10 +14,10 @@ model_name = "bn22/Mistral-7B-Instruct-v0.1-sharded"
14
  # function for loading 4-bit quantized model
15
  def load_quantized_model(model_name: str):
16
 
17
- model = HuggingFaceHub(
18
- repo_id="google/flan-ul2",
19
- model_kwargs={"temperature":0.1,
20
- "max_new_tokens":256})
21
 
22
  """
23
  :param model_name: Name or path of the model to be loaded.
 
14
  # function for loading 4-bit quantized model
15
  def load_quantized_model(model_name: str):
16
 
17
+ model = HuggingFaceHub(
18
+ repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1",
19
+ model_kwargs={"max_length": 1048, "temperature":0.2, "max_new_tokens":256, "top_p":0.95, "repetition_penalty":1.0},
20
+ )
21
 
22
  """
23
  :param model_name: Name or path of the model to be loaded.