Spaces:
Running
on
A10G
Running
on
A10G
hangzhang-nlp
commited on
Commit
·
4671ce0
1
Parent(s):
12c174c
Update app.py
Browse files
app.py
CHANGED
@@ -134,7 +134,7 @@ def gradio_answer(chatbot, chat_state, img_list, num_beams, temperature):
|
|
134 |
num_beams=1,
|
135 |
temperature=temperature,
|
136 |
max_new_tokens=240,
|
137 |
-
max_length=
|
138 |
chatbot[-1][1] = llm_message
|
139 |
print(chat_state.get_prompt())
|
140 |
print(chat_state)
|
|
|
134 |
num_beams=1,
|
135 |
temperature=temperature,
|
136 |
max_new_tokens=240,
|
137 |
+
max_length=511)[0]
|
138 |
chatbot[-1][1] = llm_message
|
139 |
print(chat_state.get_prompt())
|
140 |
print(chat_state)
|