huggingface / huggingface/diffusers
custom_pipeline not being cached
- 主要言語
- Python
- スター
- 34.5k
- フォーク
- 7.3k
- 平均マージ
- 3日 3時間
- マージ済み PR(30日)
- 91
説明
### 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
コントリビューションガイド
調査の方向性
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.
索引モデルが issue の本文から書いたものです。
評価
- 技術スタック
- python
- 領域
- machine-learning
- issue の種類
- バグ
- 難易度
- 3/5
- 見積もり時間
- 1〜2日
- 活発さ
- 停滞
- 明瞭さ
- おおむね明確
- 初心者へのやさしさ
- 35/100