commit

app.py

@@ -409,11 +409,9 @@ CHECKPOINTS = ESRGANUpscalerCheckpoints(
 )
 
 # initialize the enhancer, on the cpu
-DEVICE_CPU = torch.device("cpu")
-DTYPE = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
-enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=DEVICE_CPU, dtype=DTYPE)
-
 device = DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+DTYPE = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float32
+enhancer = ESRGANUpscaler(checkpoints=CHECKPOINTS, device=DEVICE, dtype=DTYPE)
 enhancer.to(device=DEVICE, dtype=DTYPE)
 
 # logging