zhifu gao
2024-04-08 6ddd383370c3b264c5c56a8fafb33e5cde20cfc5
setup.py
@@ -40,6 +40,7 @@
        "hydra-core>=1.3.2",
        "tensorboardX",
        "rotary_embedding_torch",
        "openai-whisper",
    ],
    # train: The modules invoked when training only.
    "train": [