[Fix] keep keys (#431)

2025-05-30 16:03:24 +08:00 · 2023-09-22 17:30:54 +08:00 · 2023-09-22 17:30:54 +08:00 · 07574fddbb
commit 07574fddbb
parent a1ea3c094a
1 changed files with 1 additions and 1 deletions
--- a/opencompass/partitioners/base.py
+++ b/opencompass/partitioners/base.py
@ -62,7 +62,7 @@ class BasePartitioner:
                        tgt_ptr[key] = {}
                    tgt_ptr = tgt_ptr[key]
                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.')

        tasks = self.partition(models,