Dev gzf deepspeed (#1833)

zhifu gao

2024-06-20 e65b1f701abca03bf3a1b5fbb200392aabd38c22

 funasr/__init__.py

@@ -1,8 +1,6 @@
"""Initialize funasr package."""

import os
import pkgutil
import importlib

dirname = os.path.dirname(__file__)
version_file = os.path.join(dirname, "version.txt")
@@ -12,6 +10,7 @@

import importlib
import pkgutil


def import_submodules(package, recursive=True):
    if isinstance(package, str):
@@ -23,7 +22,7 @@
            pass
    results = {}
    if not isinstance(package, str):
        for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + '.'):
        for loader, name, is_pkg in pkgutil.walk_packages(package.__path__, package.__name__ + "."):
            try:
                results[name] = importlib.import_module(name)
            except Exception as e:
@@ -34,9 +33,10 @@
                results.update(import_submodules(name))
    return results


import_submodules(__name__)

from funasr.auto.auto_model import AutoModel
from funasr.auto.auto_frontend import AutoFrontend

os.environ["HYDRA_FULL_ERROR"] = "1"
os.environ["HYDRA_FULL_ERROR"] = "1"