Update src/models/whisper_specrnet.py

src/models/whisper_specrnet.py

@@ -12,7 +12,7 @@ class WhisperSpecRNet(SpecRNet):
         super().__init__(input_channels=input_channels, **kwargs)
 
         self.device = kwargs["device"]
-        checkpoint = torch.load(WHISPER_MODEL_WEIGHTS_PATH)
+        checkpoint = torch.load(WHISPER_MODEL_WEIGHTS_PATH, map_location=torch.device('cpu'))
         dims = ModelDimensions(**checkpoint["dims"].__dict__)
         model = Whisper(dims)
         model = model.to(self.device)