huggingface / huggingface/diffusers

single_file_utils.py#load_single_file_model_checkpoint bugs

Ouverte
#7,889 5 commentaires 0 réactions 0 personnes assignées Voir sur GitHub
bug stale
Langage dominant
Python
Étoiles
34.5k
Forks
7.3k
Merge moyen
3 j 3 h
PR mergées (30 j)
91

Description

### Describe the bug

Loading a checkpoint from file crashes as it tries to print a warning with unexpected keys because is accessing `__name__` on incorrect target (instance instead of class)

https://github.com/huggingface/diffusers/blob/d50baf0c632342b9576a24352244c4235ce8b875/src/diffusers/loaders/single_file_utils.py

`f"Some weights of the model checkpoint were not used when initializing {vae.__name__}`
`f"Some weights of the model checkpoint were not used when initializing {unet.__name__}`
`f"Some weights of the model checkpoint were not used when initializing {controlnet.__name__}`

should be:

`[...].__class__.__name__`

### Reproduction

load a single file checkpoint with extra keys

### Logs

_No response_

### System Info

- `diffusers` version: 0.27.2
- Platform: Linux-6.8.8-1-clear-x86_64-with-glibc2.39
- Python version: 3.11.0
- PyTorch version (GPU?): 2.2.0+cu121 (True)
- Huggingface_hub version: 0.23.0
- Transformers version: 4.40.1
- Accelerate version: 0.30.0

### Who can help?

@sayakpaul

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Open src/diffusers/loaders/single_file_utils.py and inspect load_single_file_model_checkpoint, especially the warning paths for VAE, UNet, and ControlNet checkpoints. Reproduce the failure by loading a single-file checkpoint with extra keys, then verify that each warning identifies the model class without crashing.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
python, pytorch
Domaine
machine-learning
Type d'issue
Bug
Difficulté
1/5
Temps estimé
Moins d'une heure
Activité
À l'abandon
Clarté
Clairement spécifiée
Accessibilité débutants
35/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.