prithivMLmods commited on
Commit
973ae39
·
verified ·
1 Parent(s): 226d715

update app

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