Update requirements

bofenghuang · May 16, 2023 · bea1b67 · bea1b67
1 parent 41bea0f
commit bea1b67
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -49,6 +49,7 @@ dependencies = [
     "torch",
     "transformers>=4.28.0,<4.29.0",
     "wandb",
+    "tensorboardX",
 ]
 
 [project.optional-dependencies]