Skip to content

Commit 3be9074

Browse files
committed
fix for the previous fix.
1 parent 8100e90 commit 3be9074

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

modules/upscaler_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def upscale_without_tiling(model, img: Image.Image):
1717
img = np.ascontiguousarray(np.transpose(img, (2, 0, 1))) / 255
1818
img = torch.from_numpy(img).float()
1919

20-
model_weight = next(iter(model.parameters()))
20+
model_weight = next(iter(model.model.parameters()))
2121
img = img.unsqueeze(0).to(device=model_weight.device, dtype=model_weight.dtype)
2222

2323
with torch.no_grad():

0 commit comments

Comments
 (0)