Spaces:

ChrisPreston
/

diff-svc_minato_aqua

Build error

ChrisPreston commited on Feb 8, 2023

Commit

6ce0876

1 Parent(s): c16ad86

Update infer_tools/infer_tool.py

Files changed (1) hide show

infer_tools/infer_tool.py CHANGED Viewed

@@ -85,7 +85,7 @@ class Svc:
             spec_min=hparams['spec_min'], spec_max=hparams['spec_max'],
         )
         utils.load_ckpt(self.model, self.model_path, 'model', force=True, strict=True)
-        self.model.cuda()
         self.vocoder = NsfHifiGAN()
     def infer(self, in_path, key, acc, spk_id=0, use_crepe=True, singer=False):

             spec_min=hparams['spec_min'], spec_max=hparams['spec_max'],
         )
         utils.load_ckpt(self.model, self.model_path, 'model', force=True, strict=True)
+        self.model.to(self.dev)
         self.vocoder = NsfHifiGAN()
     def infer(self, in_path, key, acc, spk_id=0, use_crepe=True, singer=False):