This commit is contained in:
lyuxiang.lx
2025-01-02 12:32:43 +08:00
parent 77d8cf13a3
commit 99ab0f4fcb
2 changed files with 5 additions and 4 deletions

View File

@@ -46,6 +46,7 @@ def get_optimized_script(model, preserved_attrs=[]):
script = torch.jit.optimize_for_inference(script)
return script
def main():
args = get_args()
logging.basicConfig(level=logging.DEBUG,
@@ -57,10 +58,10 @@ def main():
try:
model = CosyVoice(args.model_dir)
except:
except Exception:
try:
model = CosyVoice2(args.model_dir)
except:
except Exception:
raise TypeError('no valid model_type!')
if not isinstance(model, CosyVoice2):

View File

@@ -58,10 +58,10 @@ def main():
try:
model = CosyVoice(args.model_dir)
except:
except Exception:
try:
model = CosyVoice2(args.model_dir)
except:
except Exception:
raise TypeError('no valid model_type!')
# 1. export flow decoder estimator