extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-04-30 06:14:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix - tasks with an initial image were not resizing the initial image to the desired dimension

Browse Source
This commit is contained in:
cmdr2 2022-09-14 11:36:55 +05:30
parent d13e08e53b
commit 3b47eb3b07
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ui/sd_internal/runtime.py
View File

@ -273,7 +273,7 @@ def mk_img(req: Request):
else: else:
handler = _img2img handler = _img2img
init_image = load_img(req.init_image) init_image = load_img(req.init_image, opt_W, opt_H)
init_image = init_image.to(device) init_image = init_image.to(device)
if device != "cpu" and precision == "autocast": if device != "cpu" and precision == "autocast":
@ -511,12 +511,15 @@ def load_model_from_config(ckpt, verbose=False):
class UserInitiatedStop(Exception): class UserInitiatedStop(Exception):
pass pass
def load_img(img_str): def load_img(img_str, w0, h0):
image = base64_str_to_img(img_str).convert("RGB") image = base64_str_to_img(img_str).convert("RGB")
w, h = image.size w, h = image.size
print(f"loaded input image of size ({w}, {h}) from base64") print(f"loaded input image of size ({w}, {h}) from base64")
if h0 is not None and w0 is not None:
h, w = h0, w0
w, h = map(lambda x: x - x % 64, (w, h)) # resize to integer multiple of 64 w, h = map(lambda x: x - x % 64, (w, h)) # resize to integer multiple of 64
image = image.resize((w, h), resample=Image.LANCZOS) image = image.resize((w, h), resample=Image.Resampling.LANCZOS)
image = np.array(image).astype(np.float32) / 255.0 image = np.array(image).astype(np.float32) / 255.0
image = image[None].transpose(0, 3, 1, 2) image = image[None].transpose(0, 3, 1, 2)
image = torch.from_numpy(image) image = torch.from_numpy(image)