Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|
74 |
-
resized_image = cropped_image.resize((1280,
|
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,
|