Spaces:
Sleeping
Sleeping
Commit
·
b7f8976
1
Parent(s):
b36323a
feat: remove num_return_sequences
Browse files- 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 |
)
|