Skip to content

S
Stable Diffusion Webui
Commit 3be90740 authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files
Options

fix for the previous fix.

parent 8100e901
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
modules/upscaler_utils.py
View file @ 3be90740
...@@ -17,7 +17,7 @@ def upscale_without_tiling(model, img: Image.Image): ...@@ -17,7 +17,7 @@ def upscale_without_tiling(model, img: Image.Image):
img = np.ascontiguousarray(np.transpose(img, (2, 0, 1))) / 255 img = np.ascontiguousarray(np.transpose(img, (2, 0, 1))) / 255
img = torch.from_numpy(img).float() img = torch.from_numpy(img).float()
model_weight = next(iter(model.parameters())) model_weight = next(iter(model.model.parameters()))
img = img.unsqueeze(0).to(device=model_weight.device, dtype=model_weight.dtype) img = img.unsqueeze(0).to(device=model_weight.device, dtype=model_weight.dtype)
with torch.no_grad(): with torch.no_grad():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment