Update app.py

app.py

@@ -209,12 +209,12 @@ def main():
     repo="stabilityai/sd-vae-ft-mse-original"
     ckpt="vae-ft-mse-840000-ema-pruned.safetensors"
     save_file(load_file(hf_hub_download(repo, ckpt), device=device),"./vae")
-    vae = ModelMixin("./vae")
+    vae = "./vae"
 
     repo="ByteDance/SDXL-Lightning"
     ckpt=f"sdxl_lightning_{step}step_unet.safetensors"
     save_file(load_file(hf_hub_download(repo, ckpt), device=device),"./unet")
-    unet = ModelMixin("./unet")
+    unet = "./unet"
     
     #repo = "SG161222/Realistic_Vision_V6.0_B1_noVAE"