Unverified Commit 50f4539b authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

push (#11400)

parent bf2e0cf7
......@@ -66,7 +66,8 @@ def set_recursively(hf_pointer, key, value, full_name, weight_type):
assert (
hf_shape == value.shape
), f"Shape of hf {key + '.' + weight_type} is {hf_shape}, but should be {value.shape} for {full_name}"
), f"Shape of hf {key + '.' + weight_type if weight_type is not None else ''} is {hf_shape}, but should be {value.shape} for {full_name}"
if weight_type == "weight":
hf_pointer.weight.data = value
elif weight_type == "weight_g":
......
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