huggingface / huggingface/diffusers

Flux Inference memory optimisation by pre-computing modulation parameters ahead of time

Abierto
#9,197 3 comentarios 2 reacciones 0 asignados Ver en GitHub
stale
Lenguaje dominante
Python
Estrellas
34.5k
Forks
7.3k
Merge medio
3 d 3 h
PR fusionados (30 d)
91

Descripción

DiffusionKit has an excellent memory optimisation for Flux where it calculates modulation parameters ahead of time
then offloads the adaLN_modulation parameters which for fp16 saves ~ 6.5 GB peak memory usage during inference.

https://github.com/argmaxinc/DiffusionKit/pull/15/

It would be nice if a similar change could be made to Diffusers.

It's especially useful for MPS users who are getting left behind with the current trend using quantisation to reduce memory usage rather than optimisation and for whom CPU offloading is not a thing.

Guía de contribución

Abrir la guía de contribución

Línea de trabajo

No se nombran archivos ni pruebas de Diffusers. Empieza leyendo el pull request enlazado de DiffusionKit y, después, localiza la inferencia de Flux y el manejo de los parámetros de modulación en Diffusers. Se considera terminado lograr la precomputación análoga y la reducción de memoria mediante offloading, preservando el comportamiento de la inferencia, también para los usuarios de MPS.

Escrito por el modelo de indexación a partir del texto del issue.

Evaluación

Stack tecnológico
python, pytorch
Área
machine-learning, performance
Tipo de issue
Refactorización
Dificultad
4/5
Tiempo estimado
3-5 días
Estado de actividad
Estancado
Claridad
Bastante claro
Aptitud para principiantes
38/100

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.