AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code 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

7
src/diffusers/pipelines/pipeline_utils.py
View File

@@ -560,10 +560,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
register_dict = {name: (None, None)}
else:
# register the config from the original module, not the dynamo compiled one
if is_compiled_module(module):
not_compiled_module = module._orig_mod
else:
not_compiled_module = module
not_compiled_module = _unwrap_model(module)
library = not_compiled_module.__module__.split(".")[0]
@@ -666,7 +663,7 @@ class DiffusionPipeline(ConfigMixin, PushToHubMixin):
# Dynamo wraps the original model in a private class.
# I didn't find a public API to get the original class.
if is_compiled_module(sub_model):
sub_model = sub_model._orig_mod
sub_model = _unwrap_model(sub_model)
model_cls = sub_model.__class__
save_method_name = None