| 17 | return is_img2img |
| 18 | |
| 19 | def ui(self, is_img2img): |
| 20 | info = gr.HTML("<p style=\"margin-bottom:0.75em\">Will upscale the image by the selected scale factor; use width and height sliders to set tile size</p>") |
| 21 | overlap = gr.Slider(minimum=0, maximum=256, step=16, label='Tile overlap', value=64, elem_id=self.elem_id("overlap")) |
| 22 | scale_factor = gr.Slider(minimum=1.0, maximum=4.0, step=0.05, label='Scale Factor', value=2.0, elem_id=self.elem_id("scale_factor")) |
| 23 | upscaler_index = gr.Radio(label='Upscaler', choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name, type="index", elem_id=self.elem_id("upscaler_index")) |
| 24 | |
| 25 | return [info, overlap, upscaler_index, scale_factor] |
| 26 | |
| 27 | def run(self, p, _, overlap, upscaler_index, scale_factor): |
| 28 | if isinstance(upscaler_index, str): |