alexanderbaikal commited on
Commit ·
fa56f53
1
Parent(s): cee1dc4
fix indentation
Browse files
app.py
CHANGED
|
@@ -10,7 +10,12 @@ def generate_video(first_frame_url, last_frame_url, prompt):
|
|
| 10 |
model_id = "Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers"
|
| 11 |
image_encoder = CLIPVisionModel.from_pretrained(model_id, subfolder="image_encoder", torch_dtype=torch.float32)
|
| 12 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
| 13 |
-
pipe = WanImageToVideoPipeline.from_pretrained(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
pipe.to("cuda")
|
| 15 |
|
| 16 |
first_frame = load_image(first_frame_url)
|
|
|
|
| 10 |
model_id = "Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers"
|
| 11 |
image_encoder = CLIPVisionModel.from_pretrained(model_id, subfolder="image_encoder", torch_dtype=torch.float32)
|
| 12 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
| 13 |
+
pipe = WanImageToVideoPipeline.from_pretrained(
|
| 14 |
+
"Wan-AI/Wan2.1-FLF2V-14B-720P-diffusers",
|
| 15 |
+
torch_dtype=torch.bfloat16,
|
| 16 |
+
low_cpu_mem_usage=True,
|
| 17 |
+
keep_in_fp32_modules=True
|
| 18 |
+
)
|
| 19 |
pipe.to("cuda")
|
| 20 |
|
| 21 |
first_frame = load_image(first_frame_url)
|