ffmpeg

游雁

2024-03-18 ee4bf7fba4c3d45aa7513e1b379ab66445fd8f53

 funasr/bin/export.py

@@ -24,8 +24,10 @@
    if kwargs.get("debug", False):
        import pdb; pdb.set_trace()

    if "device" not in kwargs:
        kwargs["device"] = "cpu"
    model = AutoModel(**kwargs)
    
    model = AutoModel(export_model=True, **kwargs)
    res = model.export(input=kwargs.get("input", None),
                       type=kwargs.get("type", "onnx"),
                       quantize=kwargs.get("quantize", False),