Update app.py

app.py

@@ -40,6 +40,7 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 if not os.path.exists(model_path) or not os.path.exists(f"{model_path}/model_real_esran") or not os.path.exists(f"{model_path}/model_rife"):
+    print(f"Model not found, downloading from Hugging Face...")
     hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir=f"{model_path}/model_real_esran")
     snapshot_download(repo_id="AlexWortega/RIFE", local_dir=f"{model_path}/model_rife")
     snapshot_download(repo_id="BestWishYsh/ConsisID-preview", local_dir=f"{model_path}")
@@ -116,10 +117,10 @@ pipe.to(device)
 
 # Enable CPU offload for the model.
 # turn on if you don't have multiple GPUs or enough GPU memory(such as H100) and it will cost more time in inference, it may also reduce the quality
-pipe.enable_model_cpu_offload()
-pipe.enable_sequential_cpu_offload()
-# pipe.vae.enable_slicing()
-# pipe.vae.enable_tiling()
+# pipe.enable_model_cpu_offload()
+# pipe.enable_sequential_cpu_offload()
+pipe.vae.enable_slicing()
+pipe.vae.enable_tiling()
 
 os.makedirs("./output", exist_ok=True)
 os.makedirs("./gradio_tmp", exist_ok=True)