lllyasviel/huggingface_guess
1
0
Fork 0
mirror of https://github.com/lllyasviel/huggingface_guess.git synced 2026-04-29 11:51:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
layerdiffusion
2024-08-03 16:19:22 -07:00
parent 0c55600114
commit aebabb94ea
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
huggingface_guess/__init__.py
View File

@@ -1,8 +1,14 @@
from huggingface_guess.detection import model_config_from_unet, unet_prefix_from_state_dict, model_config_from_diffusers_unet
def guess_repo_name(state_dict):
def guess(state_dict):
unet_key_prefix = unet_prefix_from_state_dict(state_dict)
model_config = model_config_from_unet(state_dict, unet_key_prefix, use_base_if_no_match=True)
repo_id = model_config.huggingface_repo
result = model_config_from_unet(state_dict, unet_key_prefix, use_base_if_no_match=False)
return result
def guess_repo_name(state_dict):
config = guess(state_dict)
assert config is not None
repo_id = config.huggingface_repo
return repo_id