Update app.py

app.py

@@ -41,23 +41,26 @@ handler2.setFormatter(formatter)
 root.addHandler(handler2)
 
 last_motion=None
-fps=10
-time=1
-width=448
-height=448
-device = "cuda"
 dtype = torch.float16
 result=[]
-step = 15
-accu=2
+device = "cuda"
 #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)
+base = "emilianJR/epiCRealism"
+#base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
+#vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
 #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)
 adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
 
+fasts=True
+fps=15
+time=20
+width=320
+height=640
+step = 25
+accu=4
+
+
 css="""
 input, input::placeholder {
     text-align: center !important;
@@ -217,7 +220,7 @@ def run(i,m,p1,p2,*result):
 
         return out
 
-pipe = AnimateDiffPipeline.from_pretrained(base, vae=vae, motion_adapter=adapter, torch_dtype=dtype).to(device)
+pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype).to(device)
 pipe.scheduler = DDIMScheduler(
         clip_sample=False,
         beta_start=0.00085,
@@ -229,7 +232,7 @@ pipe.scheduler = DDIMScheduler(
 #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=fasts)
 
 mp.set_start_method("spawn", force=True)