Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -33,7 +33,7 @@ step = 2
|
|
33 |
#ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
34 |
#base = "emilianJR/epiCRealism"
|
35 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
36 |
-
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
|
37 |
#unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
|
38 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
39 |
|
@@ -193,7 +193,7 @@ def run(i,m,p1,p2,*result):
|
|
193 |
|
194 |
return out
|
195 |
|
196 |
-
pipe = AnimateDiffPipeline.from_pretrained(base,
|
197 |
pipe.scheduler = EulerDiscreteScheduler(
|
198 |
beta_start=0.00085,
|
199 |
beta_end=0.012,
|
|
|
33 |
#ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
34 |
#base = "emilianJR/epiCRealism"
|
35 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
36 |
+
#vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
|
37 |
#unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
|
38 |
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
39 |
|
|
|
193 |
|
194 |
return out
|
195 |
|
196 |
+
pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype).to(device)
|
197 |
pipe.scheduler = EulerDiscreteScheduler(
|
198 |
beta_start=0.00085,
|
199 |
beta_end=0.012,
|