huggingface / huggingface/diffusers

UNet3DConditionModel does not use class conditioning

Abierto
#7,053 7 comentarios 0 reacciones 0 asignados Ver en GitHub
bug stale
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

Abrir la 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

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.