aryo100 commited on
Commit
6e0aaee
·
1 Parent(s): dd9b47f

update app

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -80,14 +80,14 @@ def generate_stream(prompt, max_new_tokens=128):
80
  model.generate(
81
  **inputs,
82
  max_new_tokens=max_new_tokens,
83
- min_new_tokens=16,
84
  do_sample=True,
85
  temperature=0.7,
86
  top_p=0.9,
87
  streamer=streamer,
88
- early_stopping=True,
89
- eos_token_id=tokenizer.eos_token_id,
90
- pad_token_id=tokenizer.pad_token_id,
91
  use_cache=True,
92
  )
93
  except Exception as e:
 
80
  model.generate(
81
  **inputs,
82
  max_new_tokens=max_new_tokens,
83
+ # min_new_tokens=16,
84
  do_sample=True,
85
  temperature=0.7,
86
  top_p=0.9,
87
  streamer=streamer,
88
+ # early_stopping=True,
89
+ # eos_token_id=tokenizer.eos_token_id,
90
+ # pad_token_id=tokenizer.pad_token_id,
91
  use_cache=True,
92
  )
93
  except Exception as e: