Segmind Vega

DOCS.md

@@ -23,7 +23,9 @@ Rescale up to 4x using [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) fro
 Each model checkpoint has a different aesthetic:
 
 * [cagliostrolab/animagine-xl-3.1](https://huggingface.co/cagliostrolab/animagine-xl-3.1): anime
+* [cyberdelia/CyberRealisticXL](https://huggingface.co/cyberdelia/CyberRealsticXL): photorealistic
 * [fluently/Fluently-XL-Final](https://huggingface.co/fluently/Fluently-XL-Final): general purpose
+* [segmind/Segmind-Vega](https://huggingface.co/segmind/Segmind-Vega): lightweight general purpose (default)
 * [SG161222/RealVisXL_V5.0](https://huggingface.co/SG161222/RealVisXL_V5.0): photorealistic
 * [stabilityai/stable-diffusion-xl-base-1.0](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0): base
 

README.md

@@ -19,31 +19,23 @@ models:
 - cyberdelia/CyberRealsticXL
 - fluently/Fluently-XL-Final
 - madebyollin/sdxl-vae-fp16-fix
+- segmind/Segmind-Vega
 - 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 animagine-xl-3.1.safetensors
+- cyberdelia/CyberRealsticXL CyberRealisticXLPlay_V1.0.safetensors
+- fluently/Fluently-XL-Final FluentlyXL-Final.safetensors
+- madebyollin/sdxl-vae-fp16-fix config.json,diffusion_pytorch_model.safetensors
+- SG161222/RealVisXL_V5.0 RealVisXL_V5.0_fp16.safetensors
 - >-
-  ai-forever/Real-ESRGAN
-  RealESRGAN_x2.pth,RealESRGAN_x4.pth
-- >-
-  cagliostrolab/animagine-xl-3.1
-  animagine-xl-3.1.safetensors
-- >-
-  cyberdelia/CyberRealsticXL
-  CyberRealisticXLPlay_V1.0.safetensors
-- >-
-  fluently/Fluently-XL-Final
-  FluentlyXL-Final.safetensors
-- >-
-  madebyollin/sdxl-vae-fp16-fix
-  config.json,diffusion_pytorch_model.safetensors
-- >-
-  SG161222/RealVisXL_V5.0
-  RealVisXL_V5.0_fp16.safetensors
+  segmind/Segmind-Vega
+  scheduler/scheduler_config.json,text_encoder/config.json,text_encoder/model.fp16.safetensors,text_encoder_2/config.json,text_encoder_2/model.fp16.safetensors,tokenizer/merges.txt,tokenizer/special_tokens_map.json,tokenizer/tokenizer_config.json,tokenizer/vocab.json,tokenizer_2/merges.txt,tokenizer_2/special_tokens_map.json,tokenizer_2/tokenizer_config.json,tokenizer_2/vocab.json,unet/config.json,unet/diffusion_pytorch_model.fp16.safetensors,vae/config.json,vae/diffusion_pytorch_model.fp16.safetensors,model_index.json
 - >-
   stabilityai/stable-diffusion-xl-base-1.0
-  scheduler/scheduler_config.json,text_encoder/config.json,text_encoder/model.fp16.safetensors,text_encoder_2/config.json,text_encoder_2/model.fp16.safetensors,tokenizer/merges.txt,tokenizer/special_tokens_map.json,tokenizer/tokenizer_config.json,tokenizer/vocab.json,tokenizer_2/merges.txt,tokenizer_2/special_tokens_map.json,tokenizer_2/tokenizer_config.json,tokenizer_2/vocab.json,unet/config.json,unet/diffusion_pytorch_model.fp16.safetensors,vae/config.json,vae/diffusion_pytorch_model.fp16.safetensors,model_index.json
+  scheduler/scheduler_config.json,text_encoder/config.json,text_encoder/model.fp16.safetensors,text_encoder_2/config.json,text_encoder_2/model.fp16.safetensors,tokenizer/merges.txt,tokenizer/special_tokens_map.json,tokenizer/tokenizer_config.json,tokenizer/vocab.json,tokenizer_2/merges.txt,tokenizer_2/special_tokens_map.json,tokenizer_2/tokenizer_config.json,tokenizer_2/vocab.json,unet/config.json,unet/diffusion_pytorch_model.fp16.safetensors,vae/config.json,vae/diffusion_pytorch_model.fp16.safetensors,vae_1_0/config.json,model_index.json
 - >-
   stabilityai/stable-diffusion-xl-refiner-1.0
   scheduler/scheduler_config.json,text_encoder_2/config.json,text_encoder_2/model.fp16.safetensors,tokenizer_2/merges.txt,tokenizer_2/special_tokens_map.json,tokenizer_2/tokenizer_config.json,tokenizer_2/vocab.json,unet/config.json,unet/diffusion_pytorch_model.fp16.safetensors,vae/config.json,vae/diffusion_pytorch_model.fp16.safetensors,model_index.json

app.py

@@ -225,7 +225,7 @@ with gr.Blocks(
                         use_refiner = gr.Checkbox(
                             elem_classes=["checkbox"],
                             label="Refiner",
-                            value=True,
+                            value=False,
                         )
 
     # Main content

lib/config.py

@@ -23,11 +23,12 @@ Config = SimpleNamespace(
         "txt2img": StableDiffusionXLPipeline,
         "img2img": StableDiffusionXLImg2ImgPipeline,
     },
-    MODEL="fluently/Fluently-XL-Final",
+    MODEL="segmind/Segmind-Vega",
     MODELS=[
         "cagliostrolab/animagine-xl-3.1",
         "cyberdelia/CyberRealsticXL",
         "fluently/Fluently-XL-Final",
+        "segmind/Segmind-Vega",
         "SG161222/RealVisXL_V5.0",
         "stabilityai/stable-diffusion-xl-base-1.0",
     ],

lib/inference.py

@@ -165,6 +165,9 @@ def generate(
         TQDM,
     )
 
+    if loader.pipe is None:
+        raise Error(f"RuntimeError: Error loading {model}")
+
     pipe = loader.pipe
     refiner = loader.refiner