Xeltron-cloud commited on
Commit
a2772bb
·
verified ·
1 Parent(s): 0eec1a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,8 +38,8 @@ async def generate_text(request: GenerateRequest):
38
  with torch.no_grad():
39
  outputs = model.generate(
40
  **inputs,
41
- max_new_tokens=request.max_new_tokens,
42
- temperature=request.temperature,
43
  do_sample=True,
44
  repetition_penalty=1.1,
45
  pad_token_id=tokenizer.eos_token_id
 
38
  with torch.no_grad():
39
  outputs = model.generate(
40
  **inputs,
41
+ # max_new_tokens=request.max_new_tokens,
42
+ # temperature=request.temperature,
43
  do_sample=True,
44
  repetition_penalty=1.1,
45
  pad_token_id=tokenizer.eos_token_id