Skip to content

GitLab

Unverified Commit 07574fdd authored by Tong Gao's avatar Tong Gao Committed by GitHub
Browse files

[Fix] keep keys (#431)

parent a1ea3c09
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
opencompass/partitioners/base.py
View file @ 07574fdd
...@@ -62,7 +62,7 @@ class BasePartitioner: ...@@ -62,7 +62,7 @@ class BasePartitioner:
tgt_ptr[key] = {} tgt_ptr[key] = {}
tgt_ptr = tgt_ptr[key] tgt_ptr = tgt_ptr[key]
tgt_ptr[key_chain[-1]] = ori_ptr[key_chain[-1]] tgt_ptr[key_chain[-1]] = ori_ptr[key_chain[-1]]
except AttributeError: except Exception:
self.logger.warning(f'Key {k} not found in config, ignored.') self.logger.warning(f'Key {k} not found in config, ignored.')
tasks = self.partition(models, tasks = self.partition(models,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment