Просмотр исходного кода

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

Johan Edstedt 1 год назад
Родитель
Сommit
370117431f
1 измененных файлов с 1 добавлено и 1 удалено
  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