mirror of
https://github.com/lllyasviel/huggingface_guess.git
synced 2026-01-26 15:29:49 +00:00
22 lines
706 B
Markdown
22 lines
706 B
Markdown
# HuggingFace Guess
|
|
|
|
A simple tool to guess an HuggingFace repo URL from a state dict.
|
|
|
|
This repo does almost the same thing as `from diffusers.loaders.single_file_utils import fetch_diffusers_config` but a bit stronger and more robust.
|
|
|
|
The main model detection logics are extracted from Diffusers and stolen from ComfyUI.
|
|
|
|
```python
|
|
import safetensors.torch as sf
|
|
import huggingface_guess
|
|
|
|
|
|
state_dict = sf.load_file('./realisticVisionV51_v51VAE.safetensors')
|
|
repo_name = huggingface_guess.guess_repo_name(state_dict)
|
|
print(repo_name)
|
|
```
|
|
|
|
The above codes will print `runwayml/stable-diffusion-v1-5`.
|
|
|
|
Then you can download (or prefetch configs) from HuggingFace to instantiate models and load weights.
|