Loosen MegaBlocks version pin (#1597)

mosaicml · Oct 17, 2024 · 10f25ae · 10f25ae
1 parent fe5890e
commit 10f25ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -119,7 +119,7 @@
 ]
 
 extra_deps['megablocks'] = [
-    'megablocks==0.6.1',
+    'megablocks<0.7.0',
     'grouped-gemm==0.1.6',
 ]