Update modules/hubert/hubert_model.py

modules/hubert/hubert_model.py

@@ -220,7 +220,7 @@ def hubert_soft(
     """
     dev = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     hubert = HubertSoft()
-    checkpoint = torch.load(path, map_location=torch.device('cpu'))
+    checkpoint = torch.load(path, map_location="cpu")
     consume_prefix_in_state_dict_if_present(checkpoint, "module.")
     hubert.load_state_dict(checkpoint)
     hubert.eval().to(dev)