add loading torch models

雾聪

2024-03-13 2ea00f89b8e9c7f7635a141fa6328d8cf1ec3030

 funasr/bin/export.py

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


    if "device" not in kwargs:
        kwargs["device"] = "cpu"
    model = AutoModel(**kwargs)
    
    res = model.export(input=kwargs.get("input", None),