huggingface / huggingface/diffusers
pipeline does not work with conversion to fp16 in the to cuda call
- Lenguaje dominante
- Python
- Estrellas
- 34.5k
- Forks
- 7.3k
- Merge medio
- 3 d 3 h
- PR fusionados (30 d)
- 91
Descripción
Hi,
i was using and doing some fine-tuning with the Pixart model.
and i found a problem that i could pin it down to:
this code works:
```
pipe2 = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-512x512", torch_dtype=torch.float16)
pipe2.to("cuda")
image = pipe2(prompt, num_inference_steps=20).images[0]
```
while this does not, it gives out just noise images :
```
pipe1 = PixArtAlphaPipeline.from_pretrained("PixArt-alpha/PixArt-XL-2-512x512")
pipe1.to("cuda", dtype=torch.float16)
image = pipe1(prompt, num_inference_steps=20).images[0]
```
i have looked through the code, and i cannot find any issues so far.
has anyone encountered a similar problem or has tips where to look further?
i am on WSL Ubuntu, diffusers==0.27.2, torch==2.2.2
Guía de contribución
Línea de trabajo
Comienza reproduciendo los dos snippets de PixArtAlphaPipeline con diffusers 0.27.2 y torch 2.2.2 en WSL Ubuntu, comparando el estado del pipeline y las salidas generadas después de cada conversión a CUDA. Se considera terminado cuando se haya identificado por qué difieren las dos rutas de conversión y la conversión de dtype produzca imágenes válidas en lugar de imágenes con ruido.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Stack tecnológico
- python, pytorch
- Área
- machine-learning
- Tipo de issue
- Error
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Estado de actividad
- Estancado
- Claridad
- Bastante claro
- Aptitud para principiantes
- 25/100