Dev gzf exp (#1665)

zhifu gao

2024-04-26 1cdb3cc28d4d89a576cc06e5cd8eb80da1f3a3aa

 funasr/bin/inference.py

@@ -14,18 +14,20 @@
            return {k: to_plain_list(v) for k, v in cfg_item.items()}
        else:
            return cfg_item
    

    kwargs = to_plain_list(cfg)
    log_level = getattr(logging, kwargs.get("log_level", "INFO").upper())

    logging.basicConfig(level=log_level)

    if kwargs.get("debug", False):
        import pdb; pdb.set_trace()
        import pdb

        pdb.set_trace()
    model = AutoModel(**kwargs)
    res = model.generate(input=kwargs["input"])
    print(res)


if __name__ == '__main__':
    main_hydra()
if __name__ == "__main__":
    main_hydra()