From c4b3f62209d92896db86a9a571944c6b52a8af8d Mon Sep 17 00:00:00 2001
From: Eitan Turok <eitan.turok@databricks.com>
Date: Wed, 21 Aug 2024 18:26:10 +0000
Subject: [PATCH] use version of composer that allows torch2.4

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 062a63a..906adb0 100644
--- a/setup.py
+++ b/setup.py
@@ -83,7 +83,8 @@
 ]
 
 extra_deps['testing'] = [
-    'mosaicml>=0.22.0',
+    # 'mosaicml>=0.23.6', # uses when released
+    'mosaicml @ git+https://github.com/mosaicml/composer.git@7c48cfc00ed5df553c947b336fee72437d2e68a7',
 ]
 
 extra_deps['all'] = list({dep for key, deps in extra_deps.items() for dep in deps if key not in {'testing'}})