StefanG2002 commited on
Commit
97fcad1
·
verified ·
1 Parent(s): ce2b463

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -22,7 +22,7 @@ tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct",
22
  class Item(BaseModel):
23
  prompt: str
24
  system_prompt: str
25
- temperature: float = 0.0
26
  max_new_tokens: int = 1048
27
  top_p: float = 0.15
28
  repetition_penalty: float = 1.0
 
22
  class Item(BaseModel):
23
  prompt: str
24
  system_prompt: str
25
+ temperature: float = 0.2
26
  max_new_tokens: int = 1048
27
  top_p: float = 0.15
28
  repetition_penalty: float = 1.0