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,
|