cleaned up

nostoslabs · Jan 16, 2024 · db08032 · db08032
1 parent 2043271
commit db08032
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/video_upscaler/main.py b/src/video_upscaler/main.py
@@ -24,9 +24,6 @@ def upscale_video(input_path: Path, output_path: Path, scale: int = 4):
     else:
         device = torch.device('cpu')
         print("Using CPU.")
-    #device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-    #device = mps_device = torch.device("mps")
-    #device = torch.device('cuda' if torch.cuda.is_available() else ('mps' if torch.backends.mps.is_available() else 'cpu'))
     upscaler = RealESRGAN(device, scale)
     upscaler.load_weights('weights/RealESRGAN_x4.pth', download=True)