Commit a5106a7c authored by brkirch's avatar brkirch

Remove extra .to(device)

parent abfa22c1
...@@ -56,7 +56,6 @@ class UpscalerScuNET(modules.upscaler.Upscaler): ...@@ -56,7 +56,6 @@ class UpscalerScuNET(modules.upscaler.Upscaler):
img = torch.from_numpy(img).float() img = torch.from_numpy(img).float()
img = img.unsqueeze(0).to(device) img = img.unsqueeze(0).to(device)
img = img.to(device)
with torch.no_grad(): with torch.no_grad():
output = model(img) output = model(img)
output = output.squeeze().float().cpu().clamp_(0, 1).numpy() output = output.squeeze().float().cpu().clamp_(0, 1).numpy()
......
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