Preload models from hub

README.md

@@ -13,6 +13,36 @@ fullWidth: false
 pinned: false
 header: mini
 license: apache-2.0
+models:
+- ai-forever/Real-ESRGAN
+- cagliostrolab/animagine-xl-3.1
+- fluently/Fluently-XL-Final
+- madebyollin/sdxl-vae-fp16-fix
+- SG161222/RealVisXL_V5.0
+- stabilityai/stable-diffusion-xl-base-1.0
+- stabilityai/stable-diffusion-xl-refiner-1.0
+preload_from_hub:
+- >-
+  ai-forever/Real-ESRGAN
+  RealESRGAN_x2.pth,RealESRGAN_x4.pth
+- >-
+  cagliostrolab/animagine-xl-3.1
+  text_encoder/model.safetensors,text_encoder_2/model.safetensors,unet/diffusion_pytorch_model.safetensors,model_index.json
+- >-
+  fluently/Fluently-XL-Final
+  text_encoder/model.safetensors,text_encoder_2/model.safetensors,unet/diffusion_pytorch_model.safetensors,model_index.json
+- >-
+  madebyollin/sdxl-vae-fp16-fix
+  diffusion_pytorch_model.safetensors,config.json
+- >-
+  SG161222/RealVisXL_V5.0
+  text_encoder/model.fp16.safetensors,text_encoder_2/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,model_index.json
+- >-
+  stabilityai/stable-diffusion-xl-base-1.0
+  text_encoder/model.fp16.safetensors,text_encoder_2/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,model_index.json
+- >-
+  stabilityai/stable-diffusion-xl-refiner-1.0
+  text_encoder_2/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,model_index.json
 ---
 
 # diffusion-xl

lib/config.py

@@ -26,9 +26,10 @@ Config = SimpleNamespace(
     },
     MODEL="fluently/Fluently-XL-Final",
     MODELS=[
+        # TODO: CyberRealisticXL once single file support is added
         "cagliostrolab/animagine-xl-3.1",
         "fluently/Fluently-XL-Final",
-        "RunDiffusion/Juggernaut-XL-v9",
+        "SG161222/RealVisXL_V5.0",
         "stabilityai/stable-diffusion-xl-base-1.0",
     ],
     VAE_MODEL="madebyollin/sdxl-vae-fp16-fix",