From 3889ba168b2fc0149fb77de1ffdadb478da15e1a Mon Sep 17 00:00:00 2001 From: Federico Berto Date: Sun, 7 May 2023 03:07:30 +0900 Subject: [PATCH] [BugFix] cannot unpack non-iterable NoneType object --- flash_attn/modules/block.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flash_attn/modules/block.py b/flash_attn/modules/block.py index 7ea2c2f..b1dcdd2 100644 --- a/flash_attn/modules/block.py +++ b/flash_attn/modules/block.py @@ -26,7 +26,7 @@ except ImportError: try: from flash_attn.ops.rms_norm import RMSNorm, dropout_add_rms_norm except ImportError: - RMSNorm, dropout_add_rms_norm = None + RMSNorm, dropout_add_rms_norm = None, None try: from flash_attn.ops.rms_norm import dropout_add_rms_norm_parallel_residual