Commit
•
688e85a
1
Parent(s):
0051b62
Local files
Browse files- gradio_demo.py +7 -0
gradio_demo.py
CHANGED
|
@@ -13,6 +13,13 @@ from CKPT_PTH import LLAVA_MODEL_PATH
|
|
| 13 |
import einops
|
| 14 |
import copy
|
| 15 |
import time
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
|
| 17 |
parser = argparse.ArgumentParser()
|
| 18 |
parser.add_argument("--opt", type=str, default='options/SUPIR_v0.yaml')
|
|
|
|
| 13 |
import einops
|
| 14 |
import copy
|
| 15 |
import time
|
| 16 |
+
from huggingface_hub import hf_hub_download
|
| 17 |
+
|
| 18 |
+
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")
|
| 19 |
+
hf_hub_download(repo_id="yushan777/SUPIR", filename="sd_xl_base_1.0_0.9vae.safetensors", local_dir="yushan777_SUPIR")
|
| 20 |
+
hf_hub_download(repo_id="yushan777/SUPIR", filename="SUPIR-v0F.ckpt", local_dir="yushan777_SUPIR")
|
| 21 |
+
hf_hub_download(repo_id="yushan777/SUPIR", filename="SUPIR-v0Q.ckpt", local_dir="yushan777_SUPIR")
|
| 22 |
+
hf_hub_download(repo_id="RunDiffusion/Juggernaut-XL-Lightning", filename="Juggernaut_RunDiffusionPhoto2_Lightning_4Steps.safetensors", local_dir="RunDiffusion_Juggernaut-XL-Lightning")
|
| 23 |
|
| 24 |
parser = argparse.ArgumentParser()
|
| 25 |
parser.add_argument("--opt", type=str, default='options/SUPIR_v0.yaml')
|