Update app.py

app.py

@@ -23,12 +23,12 @@ import jax
 import jax.numpy as jnp
 
 last_motion=None
-fps=25
-time=6
+fps=15
+time=1
 device = "cuda"
 dtype = torch.float16
 result=[]
-step = 8
+step = 2
 
 css="""
 input, input::placeholder {
@@ -189,10 +189,9 @@ ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 base = "emilianJR/epiCRealism"
 #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 
-adapter = MotionAdapter().to(device, dtype)
-adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
-pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype).to(device)
+pipe = AnimateDiffPipeline.from_pretrained(bases[base_loaded], torch_dtype=dtype).to(device)
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
+pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
 pipe.enable_free_init(method="butterworth", use_fast_sampling=False)
 
 mp.set_start_method("spawn", force=True)