Spaces:

wcy1122
/

MGM-Omni

Running on Zero

wcy1122 commited on Aug 21

Commit

2921c37

1 Parent(s): 4a382ae

fp16 cosyvoice

Files changed (1) hide show

mgm/model/language_model/mgm_tts.py CHANGED Viewed

@@ -102,7 +102,7 @@ class MGMTTSForCausalLM(Qwen3ForCausalLM, MGMTTSMetaForCausalLM, MGMTTSGeneratio
                     allow_patterns="cosyvoice2/*"
                 )
             cosyvoice_path = os.path.join(model_dir, 'cosyvoice2')
-        self.cosyvoice = CosyVoice2(cosyvoice_path, fp16=False)
     def get_model(self):
         return self.model

                     allow_patterns="cosyvoice2/*"
                 )
             cosyvoice_path = os.path.join(model_dir, 'cosyvoice2')
+        self.cosyvoice = CosyVoice2(cosyvoice_path, fp16=True)
     def get_model(self):
         return self.model