"vscode:/vscode.git/clone" did not exist on "56922cfd3dc5e10893570dc87b10bc6c10af634c"
Unverified Commit 6f40a213 authored by Zach Mueller's avatar Zach Mueller Committed by GitHub
Browse files

Fix if else and *actually* enable superfast init (#32007)

* Fix if else

* rm err raise
parent e3917064
...@@ -349,7 +349,7 @@ def check_support_param_buffer_assignment(model_to_load, state_dict, start_prefi ...@@ -349,7 +349,7 @@ def check_support_param_buffer_assignment(model_to_load, state_dict, start_prefi
return False 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", False): if not getattr(model_to_load, "_supports_param_buffer_assignment", True):
logger.debug( logger.debug(
f"{model_to_load.__class__.__name__} does not support param buffer assignment, loading will be slower" f"{model_to_load.__class__.__name__} does not support param buffer assignment, loading will be slower"
) )
......
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