Default scale 2x

README.md

@@ -14,6 +14,7 @@ pinned: true
 header: mini
 license: apache-2.0
 models:
+- ai-forever/Real-ESRGAN
 - fluently/Fluently-v4
 - Linaqruf/anything-v3-1
 - Lykon/dreamshaper-8
@@ -21,6 +22,9 @@ models:
 - runwayml/stable-diffusion-v1-5
 - SG161222/Realistic_Vision_V5.1_noVAE
 preload_from_hub:
+- >-
+  ai-forever/Real-ESRGAN
+  RealESRGAN_x2.pth,RealESRGAN_x4.pth
 - >-
   fluently/Fluently-v4
   text_encoder/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,vae/diffusion_pytorch_model.fp16.safetensors

app.py

@@ -135,11 +135,11 @@ with gr.Blocks(
                             label="Images",
                         )
                         scale = gr.Dropdown(
-                            choices=[("1x", 1), ("2x", 2), ("4x", 4)],
+                            choices=[(f"{s}x", s) for s in cfg.SCALES],
                             filterable=False,
+                            value=cfg.SCALE,
                             label="Scale",
                             min_width=200,
-                            value=1,
                         )
 
                     with gr.Row():

cli.py

@@ -24,7 +24,7 @@ def main():
     parser.add_argument("-h", "--height", type=int, metavar="INT", default=cfg.HEIGHT)
     parser.add_argument("-m", "--model", type=str, metavar="STR", default=cfg.MODEL)
     parser.add_argument("-d", "--deepcache", type=int, metavar="INT", default=cfg.DEEPCACHE_INTERVAL)
-    parser.add_argument("--scale", type=int, metavar="INT", choices=[1, 2, 4], default=1)
+    parser.add_argument("--scale", type=int, metavar="INT", choices=cfg.SCALES, default=cfg.SCALE)
     parser.add_argument("--style", type=str, metavar="STR", default=cfg.STYLE)
     parser.add_argument("--scheduler", type=str, metavar="STR", default=cfg.SCHEDULER)
     parser.add_argument("--guidance", type=float, metavar="FLOAT", default=cfg.GUIDANCE_SCALE)

config.py

@@ -51,3 +51,7 @@ INFERENCE_STEPS = 30
 DEEPCACHE_INTERVAL = 2
 
 TOME_RATIO = 0.0
+
+SCALE = 2
+
+SCALES = [1, 2, 4]

usage.md

@@ -32,6 +32,10 @@ Arrays allow you to generate different images from a single prompt. For example,
 
 Styles are prompt templates from twri's [sdxl_prompt_styler](https://github.com/twri/sdxl_prompt_styler) Comfy node. Start with a subject like "cat", pick a style, and iterate from there.
 
+### Scale
+
+Rescale up to 4x using [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN).
+
 ### Models
 
 Each model checkpoint has a different aesthetic: