From 370117431ffc5dc000fb46f6e581b74bdb2c3ff8 Mon Sep 17 00:00:00 2001 From: Johan Edstedt Date: Wed, 23 Oct 2024 11:31:04 +0200 Subject: [PATCH] silence xformers warning, should also update to torch native attn flash, but lazy --- romatch/models/transformer/layers/block.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/romatch/models/transformer/layers/block.py b/romatch/models/transformer/layers/block.py index 25488f5..a711a1f 100644 --- a/romatch/models/transformer/layers/block.py +++ b/romatch/models/transformer/layers/block.py @@ -29,7 +29,7 @@ XFORMERS_AVAILABLE = True except ImportError: - logger.warning("xFormers not available") + # logger.warning("xFormers not available") XFORMERS_AVAILABLE = False