huggingface / huggingface/diffusers

custom_pipeline not being cached

Ouverte
#7,946 1 commentaire 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

Components of custom pipelines (defined via custom_pipeline parameter) do not seem to be cached.
I have a multistep system where I'm loading all my components and then caching them explicitly using transformers.utils.move_cache.
I have an sdxl pipeline with `custom_pipeline="lpw_stable_diffusion_xl"`. The normal pipeline components seem to be cached correctly but not the components of the custom_pipeline.

I'm restoring the cached components by using the Pipeline with `local_files_only=True`, however diffusers is always trying to reach the internet for the custom_pipeline components, which it cannot, since the env where the restoration happens, doesn't have a connection.

### Reproduction

```python
StableDiffusionXLPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16,
custom_pipeline="lpw_stable_diffusion_xl"
)
```

### Logs

_No response_

### System Info

latest diffusers version from main branch

### Who can help?

@yiyixuxu @sayakpaul @DN6

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Start with StableDiffusionXLPipeline.from_pretrained using custom_pipeline="lpw_stable_diffusion_xl", then inspect how transformers.utils.move_cache handles the custom pipeline components. Reproduce the restoration with local_files_only=True and verify that the custom components load from the cache without an internet connection.

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

Évaluation

Stack technique
python
Domaine
machine-learning
Type d'issue
Bug
Difficulté
3/5
Temps estimé
1-2 jours
Activité
À l'abandon
Clarté
Plutôt claire
Accessibilité débutants
35/100

Recevez les nouvelles issues par e-mail

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