shoom013 commited on
Commit
8dc627d
·
verified ·
1 Parent(s): 567ed85

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -67,9 +67,9 @@ Settings.llm = HuggingFaceInferenceAPI(model_name="mistralai/Mistral-Nemo-Instru
67
  device_map="auto",
68
  system_prompt = system_propmpt,
69
  context_window=4096,
70
- max_new_tokens=256,
71
  # stopping_ids=[50278, 50279, 50277, 1, 0],
72
- generate_kwargs={"temperature": 0.5, "do_sample": False},
73
  # tokenizer_kwargs={"max_length": 4096},
74
  tokenizer_name="mistralai/Mistral-Nemo-Instruct-2407",
75
  )
 
67
  device_map="auto",
68
  system_prompt = system_propmpt,
69
  context_window=4096,
70
+ max_new_tokens=512,
71
  # stopping_ids=[50278, 50279, 50277, 1, 0],
72
+ generate_kwargs={"temperature": 0.6, "do_sample": False},
73
  # tokenizer_kwargs={"max_length": 4096},
74
  tokenizer_name="mistralai/Mistral-Nemo-Instruct-2407",
75
  )