iiced commited on
Commit
6db16c9
1 Parent(s): 473963a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -6,13 +6,13 @@ import uvicorn
6
 
7
  app = FastAPI()
8
 
9
- client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
10
 
11
  class Item(BaseModel):
12
  prompt: str
13
  history: list
14
  system_prompt: str
15
- temperature: float = 0.0
16
  max_new_tokens: int = 1048
17
  top_p: float = 0.15
18
  repetition_penalty: float = 1.0
 
6
 
7
  app = FastAPI()
8
 
9
+ client = InferenceClient("mistralai/Mixtral-8x22B-Instruct-v0.1")
10
 
11
  class Item(BaseModel):
12
  prompt: str
13
  history: list
14
  system_prompt: str
15
+ temperature: float = 0.2
16
  max_new_tokens: int = 1048
17
  top_p: float = 0.15
18
  repetition_penalty: float = 1.0