kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Unwrap models everywhere (#5789)

Browse Source 
more debug
This commit is contained in:
Patrick von Platen
2023-11-14 12:08:03 +01:00
committed by GitHub
parent bfe94a3993
commit 81df9c85de
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/pipelines/pipeline_utils.py
+2 -5
View File
@@ -560,10 +560,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
register_dict = {name: (None, None)} register_dict = {name: (None, None)}
else: else:
# register the config from the original module, not the dynamo compiled one # register the config from the original module, not the dynamo compiled one
if is_compiled_module(module): not_compiled_module = _unwrap_model(module)
not_compiled_module = module._orig_mod
else:
not_compiled_module = module
library = not_compiled_module.__module__.split(".")[0] library = not_compiled_module.__module__.split(".")[0]
@@ -666,7 +663,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
# Dynamo wraps the original model in a private class. # Dynamo wraps the original model in a private class.
# I didn't find a public API to get the original class. # I didn't find a public API to get the original class.
if is_compiled_module(sub_model): if is_compiled_module(sub_model):
sub_model = sub_model._orig_mod sub_model = _unwrap_model(sub_model)
model_cls = sub_model.__class__ model_cls = sub_model.__class__
save_method_name = None save_method_name = None