fix bug in _get_named_modules (#3052)

* bug fix * bug fix
huggingface · Sep 6, 2024 · e7e0181 · e7e0181
1 parent 5ad982a
commit e7e0181
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/utils/modeling.py b/src/accelerate/utils/modeling.py
@@ -660,7 +660,7 @@ def _get_named_modules(
             memo.add(module)
         yield prefix, module
         for name, sub_module in module._modules.items():
-            if module is None:
+            if sub_module is None:
                 continue
             submodule_prefix = prefix + ("." if prefix else "") + name
             yield from _get_named_modules(sub_module, memo, submodule_prefix, remove_duplicate)