Local files

gradio_demo.py

@@ -13,6 +13,13 @@ from CKPT_PTH import LLAVA_MODEL_PATH
 import einops
 import copy
 import time
+from huggingface_hub import hf_hub_download
+
+hf_hub_download(repo_id="laion/CLIP-ViT-bigG-14-laion2B-39B-b160k", filename="open_clip_pytorch_model.bin", local_dir="laion_CLIP-ViT-bigG-14-laion2B-39B-b160k")
+hf_hub_download(repo_id="yushan777/SUPIR", filename="sd_xl_base_1.0_0.9vae.safetensors", local_dir="yushan777_SUPIR")
+hf_hub_download(repo_id="yushan777/SUPIR", filename="SUPIR-v0F.ckpt", local_dir="yushan777_SUPIR")
+hf_hub_download(repo_id="yushan777/SUPIR", filename="SUPIR-v0Q.ckpt", local_dir="yushan777_SUPIR")
+hf_hub_download(repo_id="RunDiffusion/Juggernaut-XL-Lightning", filename="Juggernaut_RunDiffusionPhoto2_Lightning_4Steps.safetensors", local_dir="RunDiffusion_Juggernaut-XL-Lightning")
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--opt", type=str, default='options/SUPIR_v0.yaml')