Akshay Shinde
2025-04-12 8c336fed790bffa6805357ad3f439fac7a3c1e6a
setup.py
@@ -41,6 +41,7 @@
        "tensorboardX",
        # "rotary_embedding_torch",
        "requests",
        "modelscope",
    ],
    # train: The modules invoked when training only.
    "train": [