Bogdan Radchenko commited on
Commit
926e698
·
verified ·
1 Parent(s): 26f762e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -28,7 +28,7 @@ model.eval()
28
  class GenerateRequest(BaseModel):
29
  prompt: str
30
  max_new_tokens: int = 512
31
- temperature: float = 1.0
32
 
33
  @app.post("/generate")
34
  async def generate_text(request: GenerateRequest):
@@ -41,6 +41,8 @@ async def generate_text(request: GenerateRequest):
41
  max_new_tokens=request.max_new_tokens,
42
  temperature=request.temperature,
43
  do_sample=True,
 
 
44
  pad_token_id=tokenizer.eos_token_id
45
  )
46
 
 
28
  class GenerateRequest(BaseModel):
29
  prompt: str
30
  max_new_tokens: int = 512
31
+ temperature: float = 0.7
32
 
33
  @app.post("/generate")
34
  async def generate_text(request: GenerateRequest):
 
41
  max_new_tokens=request.max_new_tokens,
42
  temperature=request.temperature,
43
  do_sample=True,
44
+ top_p=0.95,
45
+ top_k=40,
46
  pad_token_id=tokenizer.eos_token_id
47
  )
48