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)
|