Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -128,7 +128,7 @@ def model_chat(prompt, image, is_batch=False):
|
|
128 |
prompts = [f"<s>{p} <Answer/>" for p in prompts]
|
129 |
prompt_inputs = processor.tokenizer(
|
130 |
prompts,
|
131 |
-
add_special_tokens=False,
|
132 |
return_tensors="pt",
|
133 |
padding=True
|
134 |
).to(device)
|
|
|
128 |
prompts = [f"<s>{p} <Answer/>" for p in prompts]
|
129 |
prompt_inputs = processor.tokenizer(
|
130 |
prompts,
|
131 |
+
add_special_tokens=False, # Explicitly set to False
|
132 |
return_tensors="pt",
|
133 |
padding=True
|
134 |
).to(device)
|