huggingface / huggingface/diffusers
UNet3DConditionModel does not use class conditioning
- Lenguaje dominante
- Python
- Estrellas
- 34.5k
- Forks
- 7.3k
- Merge medio
- 3 d 3 h
- PR fusionados (30 d)
- 91
Descripción
### Describe the bug
Hello guys,
I found a small inconsistency. In `UNet3DConditionModel` forward method, `class_labels` is [defined](https://github.com/huggingface/diffusers/blob/6fedbd850a282ef21009f99a394f619ed7420137/src/diffusers/models/unets/unet_3d_condition.py#L540C13-L540C25) but is never used after that.
This is different from its twin model `UNet2DConditionModel`, where it is [used](https://github.com/huggingface/diffusers/blob/6fedbd850a282ef21009f99a394f619ed7420137/src/diffusers/models/unets/unet_2d_condition.py#L992) to get an embedding which is later used by the model.
Cheers.
### Reproduction
Does not apply.
### Logs
_No response_
### System Info
Does not apply.
### Who can help?
_No response_
Guía de contribución
Línea de trabajo
Empieza en src/diffusers/models/unets/unet_3d_condition.py, en el método forward de UNet3DConditionModel, y compara cómo se gestiona class_labels con la implementación enlazada de UNet2DConditionModel. Sigue cómo se obtiene y se utiliza el embedding de clase en el modelo 2D. Se considera terminado cuando el condicionamiento de clase se gestiona de forma coherente en el modelo 3D y el comportamiento relevante está cubierto por las pruebas de modelos existentes del proyecto.
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
- Estancado
- Claridad
- Bastante claro
- Aptitud para principiantes
- 45/100