fix the bug with broken rescaling in PR

This commit is contained in:
AUTOMATIC
2022-09-30 10:38:48 +03:00
parent 8f1b315318
commit a5e7b371d6
2 changed files with 12 additions and 4 deletions

View File

@@ -60,7 +60,7 @@ class Upscaler:
break
img = self.do_upscale(img, selected_model)
if img.width != dest_w or img.height != dest_h:
img = img.resize((dest_w, dest_h), resample=LANCZOS)
img = img.resize((int(dest_w), int(dest_h)), resample=LANCZOS)
return img