Update app.py

app.py

@@ -206,13 +206,12 @@ def main():
 
     adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-2")
     vae = AutoencoderKL.from_single_file("https://huggingface.co/stabilityai/sd-vae-ft-mse-original/vae-ft-mse-840000-ema-pruned.safetensors")
-    unet = UNet3DConditionModel().to(device, dtype).load_state_dict(load_file(hf_hub_download("ByteDance/SDXL-Lightning", f"sdxl_lightning_{step}step_unet.safetensors"), device=device), strict=False)
+    unet = UNet3DConditionModel().to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
     
     repo = "ByteDance/AnimateDiff-Lightning"
     ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
     base = "black-forest-labs/FLUX.1-schnell"
     #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
-    #base = "emilianJR/epiCRealism"
     
     pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, feature_extractor=None, image_encoder=None, unet=unet, torch_dtype=dtype, token=os.getenv("hf_token")).to(device)
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")