Update xformers pin (#64)

Updates xformers pin to 0.21 as it is the only version that has the right binaries needed to run it on H100s.
mosaicml · Sep 6, 2023 · 09d7977 · 09d7977
1 parent 7d28e21
commit 09d7977
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
     'diffusers[torch]==0.19.3',
     'transformers[torch]==4.31.0',
     'wandb==0.15.4',
-    'xformers==0.0.20',
+    'xformers==0.0.21',
     'triton==2.0.0',
     'torchmetrics[image]==0.11.4',
     'clean-fid',