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