alexanderbaikal
commited on
Commit
·
cee1dc4
1
Parent(s):
839bb6e
fix indentation
Browse files
app.py
CHANGED
@@ -10,8 +10,7 @@ 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(low_cpu_mem_usage=True, model_id, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16
|
14 |
-
)
|
15 |
pipe.to("cuda")
|
16 |
|
17 |
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(low_cpu_mem_usage=True, model_id, vae=vae, image_encoder=image_encoder, torch_dtype=torch.bfloat16)
|
|
|
14 |
pipe.to("cuda")
|
15 |
|
16 |
first_frame = load_image(first_frame_url)
|