Fabrice-TIERCELIN commited on
Commit
bd6a6a5
1 Parent(s): 43c5658

Add models from SUPIR

Browse files
Files changed (1) hide show
  1. GenVideo_app.py +8 -0
GenVideo_app.py CHANGED
@@ -6,6 +6,14 @@ from diffusers.utils import export_to_video
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
  import uuid
8
  import spaces
 
 
 
 
 
 
 
 
9
 
10
  # Available adapters (replace with your actual adapter names)
11
  adapter_options = {
 
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
7
  import uuid
8
  import spaces
9
+ from huggingface_hub import hf_hub_download
10
+
11
+ 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")
12
+ hf_hub_download(repo_id="camenduru/SUPIR", filename="sd_xl_base_1.0_0.9vae.safetensors", local_dir="yushan777_SUPIR")
13
+ hf_hub_download(repo_id="camenduru/SUPIR", filename="SUPIR-v0F.ckpt", local_dir="yushan777_SUPIR")
14
+ hf_hub_download(repo_id="camenduru/SUPIR", filename="SUPIR-v0Q.ckpt", local_dir="yushan777_SUPIR")
15
+ hf_hub_download(repo_id="RunDiffusion/Juggernaut-XL-Lightning", filename="Juggernaut_RunDiffusionPhoto2_Lightning_4Steps.safetensors", local_dir="RunDiffusion_Juggernaut-XL-Lightning")
16
+
17
 
18
  # Available adapters (replace with your actual adapter names)
19
  adapter_options = {