[BugFix] cannot unpack non-iterable NoneType object

69f5f7d0 · Federico Berto · 3889ba16 · 69f5f7d0
Commit 69f5f7d0 authored May 07, 2023 by Federico Berto
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

flash_attn/models/gpt.py flash_attn/models/gpt.py +1 -1

No files found.
--- 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