Update pyproject.toml

PKU-YuanGroup · Mar 17, 2024 · 25c3e06 · 25c3e06
1 parent c066d17
commit 25c3e06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -25,7 +25,7 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-train = ["deepspeed==0.9.5", "pydantic==1.10.13"]
+train = ["deepspeed==0.9.5", "pydantic==1.10.13", "moviepy==1.0.3", "wandb==0.16.3", "tensorboard==2.14.0"]
 dev = ["mypy==1.8.0"]