From 69f5f7d0a23db4debd6888d2b21c5a6a898f8432 Mon Sep 17 00:00:00 2001 From: Federico Berto Date: Sun, 7 May 2023 03:07:44 +0900 Subject: [PATCH] [BugFix] cannot unpack non-iterable NoneType object --- flash_attn/models/gpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flash_attn/models/gpt.py b/flash_attn/models/gpt.py index 0293623..111641a 100644 --- a/flash_attn/models/gpt.py +++ b/flash_attn/models/gpt.py @@ -46,7 +46,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