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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -38,11 +38,11 @@ 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
46
  )
47
 
48
  full_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
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
46
  )
47
 
48
  full_text = tokenizer.decode(outputs[0], skip_special_tokens=True)