Unverified Commit 62aa270f authored by Zach Mueller's avatar Zach Mueller Committed by GitHub
Browse files

Disable quick init for deepspeed (#32066)

Disable via deepspeed
parent 89575b56
...@@ -344,10 +344,15 @@ def check_support_param_buffer_assignment(model_to_load, state_dict, start_prefi ...@@ -344,10 +344,15 @@ def check_support_param_buffer_assignment(model_to_load, state_dict, start_prefi
as when loading in empty weights) by first checking as when loading in empty weights) by first checking
if the model explicitly disables it, then by ensuring that the state dict keys if the model explicitly disables it, then by ensuring that the state dict keys
are a subset of the model's parameters. are a subset of the model's parameters.
Note: We fully disable this if we are using `deepspeed`
""" """
if len([key for key in state_dict if key.startswith(start_prefix)]) == 0: if len([key for key in state_dict if key.startswith(start_prefix)]) == 0:
return False return False
if is_deepspeed_zero3_enabled():
return False
# Some models explicitly do not support param buffer assignment # Some models explicitly do not support param buffer assignment
if not getattr(model_to_load, "_supports_param_buffer_assignment", True): if not getattr(model_to_load, "_supports_param_buffer_assignment", True):
logger.debug( logger.debug(
......
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