khang119966 commited on
Commit
1296618
·
verified ·
1 Parent(s): a21dba0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -396,7 +396,7 @@ def generate_video(image, prompt, max_tokens):
396
  print(image)
397
  pixel_values, target_aspect_ratio = load_image(image, max_num=6)
398
  pixel_values = pixel_values.to(torch.bfloat16).cuda()
399
- generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 1, repetition_penalty=2.5)
400
  response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
401
  attention_visualize=True,last_visualize_layers=7,raw_image_path=image,target_aspect_ratio=target_aspect_ratio)
402
 
 
396
  print(image)
397
  pixel_values, target_aspect_ratio = load_image(image, max_num=6)
398
  pixel_values = pixel_values.to(torch.bfloat16).cuda()
399
+ generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
400
  response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
401
  attention_visualize=True,last_visualize_layers=7,raw_image_path=image,target_aspect_ratio=target_aspect_ratio)
402