bump torch (#146)

databricks · Aug 21, 2024 · c1c5f22 · c1c5f22
1 parent 6bfbc42
commit c1c5f22
Showing 1 changed file with 3 additions and 4 deletions.
diff --git 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://[email protected]/eitanturok/stk.git@bump-version',
+    'stanford-stk>=0.7.1',
 ]
 
 extra_deps = {}
 
 extra_deps['gg'] = [
-    'grouped_gemm @ git+https://[email protected]/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'}})