Kun Zou
2024-12-21 b5ad7c81be2e24f255cac3d0ef0037bf88228366
setup.py
@@ -41,6 +41,7 @@
        "tensorboardX",
        # "rotary_embedding_torch",
        "requests",
        "modelscope",
    ],
    # train: The modules invoked when training only.
    "train": [