huggingface / huggingface/diffusers

Segmentation fault (core dumped) when sdxl-turbo inference with torch 2.2.1+cu118

オープン
#9,889 コメント 4 件 リアクション 0 件 担当者 0 名 GitHub で見る
bug stale
主要言語
Python
スター
34.5k
フォーク
7.3k
平均マージ
3日 3時間
マージ済み PR(30日)
91

説明

### Describe the bug

When sdxl-turbo inferencing, I encountered Segmentation fault (core dumped) after loading the model. And this will happen with torch 2.2.1+cu118(xformers0.0.25+cu118), and it will not happen with torch 2.0.1+cu118(xformers0.0.20+cu118). However, I need to run under torch 2.2.1, so could anyone help me to solve this issue? Thanx!

### Reproduction

code:
pipe = AutoPipelineForText2Image.from_pretrained( cache_dir, torch_dtype=torch.float16, variant="fp16")

pipe.to("cuda")

prompt = "A cinematic shot of a baby racoon wearing an intricate italian priest robe."

image = pipe(prompt=prompt, num_inference_steps=1, guidance_scale=0.0).images[0]

### Logs

```shell
Loading pipeline components...: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 7/7 [00:01<00:00, 4.47it/s]
0%| | 0/1 [00:00

コントリビューションガイド

コントリビューションガイドを開く

調査の方向性

Start by reproducing the provided AutoPipelineForText2Image example with Python 3.10, torch 2.2.1+cu118, diffusers 0.26.3, and the reported xformers version. Compare it with the torch 2.0.1 environment and inspect the failure during the one-step CUDA inference; done means the segmentation fault is explained and a verified resolution or compatibility requirement is identified.

索引モデルが issue の本文から書いたものです。

評価

技術スタック
python, pytorch
領域
machine-learning
issue の種類
バグ
難易度
4/5
見積もり時間
3〜5日
活発さ
停滞
明瞭さ
説明が足りない
初心者へのやさしさ
25/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。