Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -144,7 +144,7 @@ tokenizer = AutoTokenizer.from_pretrained("khang119966/Vintern-1B-v3_5-explainab
|
|
144 |
|
145 |
@spaces.GPU
|
146 |
def generate_video(image, prompt, max_tokens):
|
147 |
-
pixel_values, target_aspect_ratio = load_image(
|
148 |
generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
|
149 |
response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
|
150 |
attention_visualize=True,last_visualize_layers=7,raw_image_path=test_image,target_aspect_ratio=target_aspect_ratio)
|
|
|
144 |
|
145 |
@spaces.GPU
|
146 |
def generate_video(image, prompt, max_tokens):
|
147 |
+
pixel_values, target_aspect_ratio = load_image(image, max_num=6).to(torch.bfloat16).cuda()
|
148 |
generation_config = dict(max_new_tokens= int(max_tokens), do_sample=False, num_beams = 3, repetition_penalty=2.5)
|
149 |
response, query = model.chat(tokenizer, pixel_values, '<image>\n'+prompt, generation_config, return_history=False, \
|
150 |
attention_visualize=True,last_visualize_layers=7,raw_image_path=test_image,target_aspect_ratio=target_aspect_ratio)
|