From 81011395635acbe45613e53ac621578adf7dbf5c Mon Sep 17 00:00:00 2001
From: Carlos Garcia Jurado Suarez <carlosg@uw.edu>
Date: Tue, 13 Feb 2024 10:32:06 -0800
Subject: [PATCH] Add tensorboardX dependency

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 80af5e3..b21d605 100644
--- 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"]