choco-conoz commited on
Commit
b7f8976
·
1 Parent(s): b36323a

feat: remove num_return_sequences

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -81,7 +81,7 @@ def main():
81
  # user_prompt, tokenize=False, add_generation_prompt=True)
82
  outputs = processor(user_prompt,
83
  max_new_tokens=256,
84
- num_return_sequences=1,
85
  temperature=0.6,
86
  top_p=0.9,
87
  )
 
81
  # user_prompt, tokenize=False, add_generation_prompt=True)
82
  outputs = processor(user_prompt,
83
  max_new_tokens=256,
84
+ # num_return_sequences=1,
85
  temperature=0.6,
86
  top_p=0.9,
87
  )