Update app.py

app.py

@@ -23,17 +23,17 @@ import jax
 import jax.numpy as jnp
 
 last_motion=None
-fps=20
+fps=15
 time=3
-width=480
-height=480
+width=576
+height=1024
 device = "cuda"
 dtype = torch.float16
 result=[]
-step = 8
+step = 4
 accu=7.5
-repo = "ByteDance/AnimateDiff-Lightning"
-ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
+#repo = "ByteDance/AnimateDiff-Lightning"
+#ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 #base = "emilianJR/epiCRealism"
 base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
@@ -135,8 +135,6 @@ def Piper(image,positive,negative,motion):
     global last_motion
     global ip_loaded
 
-    #pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
-
     if last_motion != motion:
         pipe.unload_lora_weights()
         if motion != "":
@@ -203,10 +201,10 @@ pipe.scheduler = DDIMScheduler(
         timestep_spacing="trailing",
         steps_offset=1
 )
-pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
+#pipe.unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device=device), strict=False)
 pipe.load_ip_adapter("h94/IP-Adapter", subfolder="models", weight_name="ip-adapter_sd15.bin")
 pipe.enable_vae_slicing()
-pipe.enable_free_init(method="butterworth", use_fast_sampling=False)
+pipe.enable_free_init(method="butterworth", use_fast_sampling=True)
 
 mp.set_start_method("spawn", force=True)