multimodalart HF staff commited on
Commit
0328b82
1 Parent(s): 0b7d3bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,8 +70,8 @@ def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guida
70
  temp = int(duration * multiplier) + 1
71
  torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
72
  if(image):
73
- cropped_image = center_crop(image, 1280, 720)
74
- resized_image = cropped_image.resize((1280, 720))
75
  with torch.no_grad(), torch.cuda.amp.autocast(enabled=True, dtype=torch_dtype):
76
  frames = model.generate_i2v(
77
  prompt=prompt,
 
70
  temp = int(duration * multiplier) + 1
71
  torch_dtype = torch.bfloat16 if MODEL_DTYPE == "bf16" else torch.float32
72
  if(image):
73
+ cropped_image = center_crop(image, 1280, 768)
74
+ resized_image = cropped_image.resize((1280, 768))
75
  with torch.no_grad(), torch.cuda.amp.autocast(enabled=True, dtype=torch_dtype):
76
  frames = model.generate_i2v(
77
  prompt=prompt,