Unverified Commit 38fe728d authored by Andrzej Kotłowski's avatar Andrzej Kotłowski Committed by GitHub
Browse files

[Bugfix] Fix QKVCrossParallelLinear::sync_weight_attrs for PyTorch compile (#17844)


Signed-off-by: default avatarAndrzej Kotłowski <akotlowski@habana.ai>
parent 82e7f9bb
......@@ -1425,8 +1425,8 @@ class QKVCrossParallelLinear(LinearBase):
):
missing_attrs_dict = {
k: getattr(src_param, k)
for k in (set(src_param.__dict__.keys()) -
set(tgt_param.__dict__.keys()))
for k in (set(vars(src_param).keys()) -
set(vars(tgt_param).keys()))
}
# TODO(Isotr0py): handle bitsandbytes 8bit
use_bitsandbytes_4bit = getattr(src_param, "use_bitsandbytes_4bit",
......
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