Yaron Koresh commited on
Commit
13c45fe
·
verified ·
1 Parent(s): 6d9be72

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -137,10 +137,11 @@ scheduler2 = DDIMScheduler.from_pretrained(
137
  pipe = AnimateDiffPipeline.from_pretrained(
138
  base,
139
  motion_adapter=motion_adapter,
140
- vae=vae,
141
- scheduler=scheduler2,
142
  torch_dtype=torch.float16
143
- ).to(device)
 
 
 
144
 
145
  pipe.enable_vae_slicing()
146
  pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")
 
137
  pipe = AnimateDiffPipeline.from_pretrained(
138
  base,
139
  motion_adapter=motion_adapter,
 
 
140
  torch_dtype=torch.float16
141
+ )
142
+
143
+ pipe.scheduler = scheduler2
144
+ pipe.vae = vae
145
 
146
  pipe.enable_vae_slicing()
147
  pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")