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

update app

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