alexanderbaikal commited on
Commit
fa56f53
·
1 Parent(s): cee1dc4

fix indentation

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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(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)
 
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)