prithivMLmods commited on
Commit
97ed157
·
verified ·
1 Parent(s): f9b15c4

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -292,8 +292,8 @@ def generate_image(model_name: str, text: str, image: Image.Image,
292
  text=[prompt_full],
293
  images=[image],
294
  return_tensors="pt",
295
- padding=True,
296
- truncation=False, # Disabled truncation
297
  ).to(device)
298
 
299
  streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
 
292
  text=[prompt_full],
293
  images=[image],
294
  return_tensors="pt",
295
+ padding=True
296
+ #truncation=False, # Disabled truncation
297
  ).to(device)
298
 
299
  streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)