Parcourir la source

silence xformers warning, should also update to torch native attn flash, but lazy

Johan Edstedt il y a 1 an
Parent
commit
370117431f
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      romatch/models/transformer/layers/block.py

+ 1 - 1
romatch/models/transformer/layers/block.py

@@ -29,7 +29,7 @@ try:
 
     XFORMERS_AVAILABLE = True
 except ImportError:
-    logger.warning("xFormers not available")
+    # logger.warning("xFormers not available")
     XFORMERS_AVAILABLE = False