|
state_dict = {k: v.to(device="meta", dtype=v.dtype) for k, v in state_dict.items()}
|
|
state_dict = {k: v.to(device="meta", dtype=v.dtype) for k, v in state_dict.items()}
|
|
original(module, state_dict, strict=strict)
|
|
original(module, state_dict, strict=strict)
|