yo

mosaicml · Aug 22, 2024 · 23566ed · 23566ed
1 parent 89532bd
commit 23566ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py
@@ -180,7 +180,8 @@ def _write_table(table_tag: str, table_contents: str):
 
 
 def _main():
-    python_pytorch_versions = [('3.11', '2.4.0'), ('3.11', '2.3.1'), ('3.11', '2.2.2')]
+    #python_pytorch_versions = [('3.11', '2.4.0'), ('3.11', '2.3.1'), ('3.11', '2.2.2')]
+    python_pytorch_versions = [('3.11', '2.3.1'), ('3.11', '2.2.2')]
     cuda_options = [True, False]
     stages = ['pytorch_stage']
     interconnects = ['mellanox', 'EFA']  # mellanox is default, EFA needed for AWS