Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -196,7 +196,7 @@ ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
|
196 |
base = "emilianJR/epiCRealism"
|
197 |
#base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
198 |
|
199 |
-
pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter,
|
200 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
|
201 |
pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
|
202 |
pipe.enable_free_init(method="butterworth", use_fast_sampling=False)
|
|
|
196 |
base = "emilianJR/epiCRealism"
|
197 |
#base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
198 |
|
199 |
+
pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
|
200 |
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
|
201 |
pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
|
202 |
pipe.enable_free_init(method="butterworth", use_fast_sampling=False)
|