Add tensorboardX dependency

AutoResearch · Feb 13, 2024 · 8101139 · 8101139
1 parent 9ba3db8
commit 8101139
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,6 +51,7 @@ pipelines = [
     "peft>=0.8.2",
     "trl>=0.7.10",
     "datasets",
+    "tensorboardX",
 ]
 # NOTE: When updating dependencies, in particular cuda/azure ml, make sure to update the azureml/conda.yaml too
 azure = ["azureml-core", "azureml-mlflow"]