Xeltron-cloud commited on
Commit
1a18c90
·
verified ·
1 Parent(s): 12376d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -42,7 +42,9 @@ async def generate_text(request: GenerateRequest):
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)
 
42
  temperature=request.temperature,
43
  do_sample=True,
44
  repetition_penalty=1.1,
45
+ eos_token_id=tokenizer.eos_token_id,
46
+ pad_token_id=tokenizer.eos_token_id,
47
+ use_cache=False
48
  )
49
 
50
  full_text = tokenizer.decode(outputs[0], skip_special_tokens=True)