From c1c5f2222967a063f48a9676e621bf145d191374 Mon Sep 17 00:00:00 2001
From: Eitan Turok <150733043+eitanturok@users.noreply.github.com>
Date: Wed, 21 Aug 2024 15:07:09 -0400
Subject: [PATCH] bump torch (#146)

---
 setup.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 906adb0..7596bd8 100644
--- a/setup.py
+++ b/setup.py
@@ -64,13 +64,13 @@
     'packaging>=21.3.0,<24.2',
     'torch>=2.4.0,<2.4.1',
     'triton>=2.1.0',
-    'stanford-stk @ git+https://git@github.com/eitanturok/stk.git@bump-version',
+    'stanford-stk>=0.7.1',
 ]
 
 extra_deps = {}
 
 extra_deps['gg'] = [
-    'grouped_gemm @ git+https://git@github.com/eitanturok/grouped_gemm.git@bump-version',
+    'grouped_gemm>=0.1.6',
 ]
 
 extra_deps['dev'] = [
@@ -83,8 +83,7 @@
 ]
 
 extra_deps['testing'] = [
-    # 'mosaicml>=0.23.6', # uses when released
-    'mosaicml @ git+https://github.com/mosaicml/composer.git@7c48cfc00ed5df553c947b336fee72437d2e68a7',
+    'mosaicml>=0.23.6',
 ]
 
 extra_deps['all'] = list({dep for key, deps in extra_deps.items() for dep in deps if key not in {'testing'}})