Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -214,7 +214,8 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 214 |
text=[prompt_full],
|
| 215 |
images=[image],
|
| 216 |
return_tensors="pt",
|
| 217 |
-
padding=True
|
|
|
|
| 218 |
).to(device)
|
| 219 |
|
| 220 |
streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
|
|
|
|
| 214 |
text=[prompt_full],
|
| 215 |
images=[image],
|
| 216 |
return_tensors="pt",
|
| 217 |
+
padding=True,
|
| 218 |
+
truncation=True, max_length=MAX_INPUT_TOKEN_LENGTH
|
| 219 |
).to(device)
|
| 220 |
|
| 221 |
streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
|