We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8100e90 commit 3be9074Copy full SHA for 3be9074
modules/upscaler_utils.py
@@ -17,7 +17,7 @@ def upscale_without_tiling(model, img: Image.Image):
17
img = np.ascontiguousarray(np.transpose(img, (2, 0, 1))) / 255
18
img = torch.from_numpy(img).float()
19
20
- model_weight = next(iter(model.parameters()))
+ model_weight = next(iter(model.model.parameters()))
21
img = img.unsqueeze(0).to(device=model_weight.device, dtype=model_weight.dtype)
22
23
with torch.no_grad():
0 commit comments