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