diff --git a/README.md b/README.md index d5b8a6b6..b7caf0f8 100644 --- a/README.md +++ b/README.md @@ -4,7 +4,7 @@ Stable Diffusion WebUI Forge is a platform on top of [Stable Diffusion WebUI](ht The name "Forge" is inspired from "Minecraft Forge". This project is aimed at becoming SD WebUI's Forge. -Forge is currently based on SD-WebUI 1.10 at [this commit](https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/8b3d98c5a580c3c72e82d03fdab2b643bf9a8edd). +Forge is currently based on SD-WebUI 1.10.1 at [this commit](https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/82a973c04367123ae98bd9abdf80d9eda9b910e2). # Installing Forge diff --git a/modules/upscaler_utils.py b/modules/upscaler_utils.py index 5ecbbed9..a8408f05 100644 --- a/modules/upscaler_utils.py +++ b/modules/upscaler_utils.py @@ -41,7 +41,7 @@ def upscale_pil_patch(model, img: Image.Image) -> Image.Image: """ param = torch_utils.get_param(model) - with torch.no_grad(): + with torch.inference_mode(): tensor = pil_image_to_torch_bgr(img).unsqueeze(0) # add batch dimension tensor = tensor.to(device=param.device, dtype=param.dtype) with devices.without_autocast():