AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

debug

Browse Source
This commit is contained in:
sayakpaul
2023-11-14 10:53:45 +05:30
parent 1fd3378282
commit 4135414907
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/utils/state_dict_utils.py
View File

@@ -210,6 +210,7 @@ def convert_state_dict_to_diffusers(state_dict, original_type=None, **kwargs):
if original_type not in DIFFUSERS_STATE_DICT_MAPPINGS.keys():
raise ValueError(f"Original type {original_type} is not supported")
print(f"*******From state_dict_utils.py: {original_type}*******")
mapping = DIFFUSERS_STATE_DICT_MAPPINGS[original_type]
return convert_state_dict(state_dict, mapping)