ChatTTS-Forge

Sleeping

zhzluke96 commited on Jun 10

Commit

db7cbc6

•

1 Parent(s): ebc4336

update

Files changed (1) hide show

modules/Enhancer/ResembleEnhance.py CHANGED Viewed

@@ -37,7 +37,7 @@ class ResembleEnhance:
         enhancer = Enhancer(hparams)
         state_dict = torch.load(
             Path(MODELS_DIR) / "resemble-enhance" / "mp_rank_00_model_states.pt",
-            map_location="cpu",
         )["module"]
         enhancer.load_state_dict(state_dict)
         enhancer.to(self.device).eval()

         enhancer = Enhancer(hparams)
         state_dict = torch.load(
             Path(MODELS_DIR) / "resemble-enhance" / "mp_rank_00_model_states.pt",
+            map_location=self.device,
         )["module"]
         enhancer.load_state_dict(state_dict)
         enhancer.to(self.device).eval()