zhifu gao
2024-04-16 a65016e23e6c829d61c63a68e27b24abf86e926a
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": [