Commit a5106a7c authored by brkirch's avatar brkirch
Browse files

Remove extra .to(device)

parent abfa22c1

modules/scunet_model.py

+0 −1
Original line number Diff line number Diff line
@@ -56,7 +56,6 @@ class UpscalerScuNET(modules.upscaler.Upscaler): 
        img = torch.from_numpy(img).float()

        img = img.unsqueeze(0).to(device)



        img = img.to(device)

        with torch.no_grad():

            output = model(img)

        output = output.squeeze().float().cpu().clamp_(0, 1).numpy()

CRA Git | Maintained and supported by SUSTech CRA and CCSE