huggingface / huggingface/diffusers
Incompatibility between FlashAttention and ERNIE-Image
Nadie ha tomado este issue todavía.
- Lenguaje dominante
- Python
- Estrellas
- 34.5k
- Forks
- 7.3k
- Merge medio
- 3 d 3 h
- PR fusionados (30 d)
- 91
Descripción
Hello,
When I enable FlashAttention backend on an ErnieImagePipeline, it crashes with following error:
ValueError: attn_mask is not supported for flash-attn 2.
But same Python code (see below) doesn't crash with ZImagePipeline and Flux2KleinPipeline.
Reproduction
pipeline.transformer.set_attention_backend("flash")
System Info
- OS: Windows
- Python: 3.13.11
- Diffusers: 9c4e201dd1cf7d16449c3f4e9affc556d7bc3404
- FlashAttention: 2.8.3+cu130torch2.10
- SDNQ: 0.1.6
Guía de contribución
Primeros pasos
- Lee el issue completo y luego la guía de contribución del proyecto.
- Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
- Haz un fork del repositorio y trabaja en una rama.
- Abre un pull request que haga referencia al número del issue.
Línea de trabajo
Empieza con ErnieImagePipeline y su configuración de atención del transformer; después compara el manejo de las máscaras de atención con ZImagePipeline y Flux2KleinPipeline. Reproduce el fallo mostrado de set_attention_backend("flash") en el entorno indicado; se considera terminado cuando ERNIE-Image puede usar el backend flash sin el error de attn_mask.
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
- 3/5
- Tiempo estimado
- 1-2 días
- Estado de actividad
- Tranquilo
- Claridad
- Bastante claro
- Aptitud para principiantes
- 55/100