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": [