Spaces:
Running
on
Zero
Running
on
Zero
Update vlm.py
Browse files
vlm.py
CHANGED
@@ -135,7 +135,7 @@ def stream_response(messages: list[dict]):
|
|
135 |
"""
|
136 |
# Generate model's response
|
137 |
inputs = processor.apply_chat_template(
|
138 |
-
|
139 |
add_generation_prompt=True,
|
140 |
tokenize=True,
|
141 |
return_dict=True,
|
|
|
135 |
"""
|
136 |
# Generate model's response
|
137 |
inputs = processor.apply_chat_template(
|
138 |
+
messages,
|
139 |
add_generation_prompt=True,
|
140 |
tokenize=True,
|
141 |
return_dict=True,
|