Skip to content

Commit

Permalink
make it possible to use fractional values for SD upscale.
Browse files Browse the repository at this point in the history
  • Loading branch information
AUTOMATIC1111 committed Jan 2, 2023
1 parent ef27a18 commit 4dbde22
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions modules/upscaler.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,10 +53,10 @@ def __init__(self, create_dirs=False):
def do_upscale(self, img: PIL.Image, selected_model: str):
return img

def upscale(self, img: PIL.Image, scale: int, selected_model: str = None):
def upscale(self, img: PIL.Image, scale, selected_model: str = None):
self.scale = scale
dest_w = img.width * scale
dest_h = img.height * scale
dest_w = int(img.width * scale)
dest_h = int(img.height * scale)

for i in range(3):
shape = (img.width, img.height)
Expand Down
2 changes: 1 addition & 1 deletion scripts/sd_upscale.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ def show(self, is_img2img):
def ui(self, is_img2img):
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>")
overlap = gr.Slider(minimum=0, maximum=256, step=16, label='Tile overlap', value=64)
scale_factor = gr.Slider(minimum=1, maximum=4, step=1, label='Scale Factor', value=2)
scale_factor = gr.Slider(minimum=1.0, maximum=4.0, step=0.05, label='Scale Factor', value=2.0)
upscaler_index = gr.Radio(label='Upscaler', choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name, type="index")

return [info, overlap, upscaler_index, scale_factor]
Expand Down

0 comments on commit 4dbde22

Please sign in to comment.