@@ -19,11 +19,14 @@ Original script with Gradio UI was written by a kind anonymous user. This is a m
- Loopback
- X/Y plot
- Textual Inversion
- Resizing options
- Extras tab with:
- GFPGAN, neural network that fixes faces
- RealESRGAN, neural network upscaler
- ESRGAN, neural network with a lot of third party models
- Resizing aspect ratio options
- Sampling method selection
- Interrupt processing at any time
- 4GB videocard support
- Option to use GFPGAN
- Correct seeds for batches
- Prompt length validation
- Generation parameters added as text to PNG
...
...
@@ -49,6 +52,9 @@ can obtain it from the following places:
You optionally can use GPFGAN to improve faces, then you'll need to download the model from [here](https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth).
To use ESRGAN models, put them into ESRGAN directory in the same location as webui.py. A file will be loaded
as model if it has .pth extension. Grab models from the [Model Database](https://upscale.wiki/wiki/Model_Database).
@@ -28,6 +28,7 @@ parser.add_argument("--always-batch-cond-uncond", action='store_true', help="a w
parser.add_argument("--unload-gfpgan",action='store_true',help="unload GFPGAN every time after processing images. Warning: seems to cause memory leaks")
parser.add_argument("--precision",type=str,help="evaluate at this precision",choices=["full","autocast"],default="autocast")
parser.add_argument("--share",action='store_true',help="use share=True for gradio and make the UI accessible through their site (doesn't work for me but you might have better luck)")
parser.add_argument("--esrgan-models-path",type=str,help="path to directory with ESRGAN models",default=os.path.join(script_path,'ESRGAN'))
cmd_opts=parser.parse_args()
cpu=torch.device("cpu")
...
...
@@ -79,7 +80,8 @@ class Options:
"font":OptionInfo("arial.ttf","Font for image grids that have text"),
"enable_emphasis":OptionInfo(True,"Use (text) to make model pay more attention to text text and [text] to make it pay less attention"),
"save_txt":OptionInfo(False,"Create a text file next to every image with generation parameters."),
"ESRGAN_tile":OptionInfo(192,"Tile size for ESRGAN upscaling. 0 = no tiling.",gr.Slider,{"minimum":0,"maximum":512,"step":16}),
"ESRGAN_tile_overlap":OptionInfo(8,"Tile overlap, in pixels for ESRGAN upscaling. Low values = visible seam.",gr.Slider,{"minimum":0,"maximum":48,"step":1}),