SSahas commited on
Commit
e8f33f0
·
verified ·
1 Parent(s): db45a40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def response_generator(prompt):
17
  print(input_text)
18
  input_ids = tokenizer(input_text, padding=True, return_tensors="pt")
19
  output_ids = model.generate(input_ids=input_ids['input_ids'], generation_config=GenerationConfig(
20
- max_new_tokens=65, pad_token_id=50256))
21
  output = tokenizer.decode(
22
  output_ids[0][input_ids['input_ids'].shape[1]:], skip_special_tokens=True)
23
 
 
17
  print(input_text)
18
  input_ids = tokenizer(input_text, padding=True, return_tensors="pt")
19
  output_ids = model.generate(input_ids=input_ids['input_ids'], generation_config=GenerationConfig(
20
+ max_new_tokens=30, pad_token_id=50256))
21
  output = tokenizer.decode(
22
  output_ids[0][input_ids['input_ids'].shape[1]:], skip_special_tokens=True)
23