Didier commited on
Commit
5d69ad6
·
verified ·
1 Parent(s): 822be23

Update vlm.py

Browse files
Files changed (1) hide show
  1. vlm.py +1 -1
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
- resulting_messages,
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,