# huggingface_guess

**Repository Path**: ai-aigc/huggingface_guess

## Basic Information

- **Project Name**: huggingface_guess
- **Description**: https://github.com/lllyasviel/huggingface_guess.git
- **Primary Language**: Unknown
- **License**: GPL-3.0
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-02-11
- **Last Updated**: 2025-02-11

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 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.