From 4c3965c216be11b04697b7498c2627a51dda18fd Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Thu, 20 Jun 2024 23:41:55 -0400 Subject: [PATCH] run tests --- setup.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index a836aec27f..bf71bdaf73 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,8 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.23.2,<0.24', + 'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.23.2,<0.25', + 'mosaicml@git+ssh://git@github.com/mvpatel2000/composer.git@mvpatel2000/test-patch#egg=mosaicml' 'mlflow>=2.13.2,<2.14', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.40,<4.41', @@ -92,14 +93,14 @@ ] extra_deps['databricks'] = [ - 'mosaicml[databricks]>=0.23.2,<0.24', + 'mosaicml[databricks]>=0.23.2,<0.25', 'databricks-sql-connector>=3,<4', 'databricks-connect==14.1.0', 'lz4>=4,<5', ] extra_deps['tensorboard'] = [ - 'mosaicml[tensorboard]>=0.23.2,<0.24', + 'mosaicml[tensorboard]>=0.23.2,<0.25', ] # Flash 2 group kept for backwards compatibility @@ -110,7 +111,7 @@ extra_deps['gpu'] = copy.deepcopy(extra_deps['gpu-flash2']) extra_deps['peft'] = [ - 'mosaicml[peft]>=0.23.2,<0.24', + 'mosaicml[peft]>=0.23.2,<0.25', ] extra_deps['openai'] = [