ClearVoice-SR

Running on Zero

alibabasglab commited on Jan 14

Commit

950bc2c

verified ·

1 Parent(s): e7b2657

Update networks.py

Files changed (1) hide show

networks.py CHANGED Viewed

@@ -165,6 +165,7 @@ class SpeechModel:
             elif self.print: print(f'{key} not loaded')
         model.load_state_dict(state)
     def decode(self):
         """
         Decodes the input audio data using the loaded model and ensures the output matches the original audio length.

             elif self.print: print(f'{key} not loaded')
         model.load_state_dict(state)
+    @spaces.GPU
     def decode(self):
         """
         Decodes the input audio data using the loaded model and ensures the output matches the original audio length.