Las GPU GeForce RTX de NVIDIA obtienen optimización para nuevos modelos de generación de imágenes con IA FLUX.2

Black Forest Labs anunció FLUX.2, una nueva familia de modelos de generación de imágenes capaces de crear imágenes fotorrealistas de 4 megapíxeles localmente en PC con GeForce RTX. 

Las imágenes generadas por FLUX.2 representan un salto significativo en la fidelidad visual, con iluminación realista, física precisa y hasta 4 veces la resolución de los modelos estándar, eliminando el “aspecto de IA”. Aunque la calidad visual es un gran avance, el tamaño de estos modelos puede llevar el hardware de consumo al límite. Para resolver esto, NVIDIA trabajó con Black Forest Labs y ComfyUI para ofrecer optimizaciones importantes en el lanzamiento:

• Cuantización FP8: NVIDIA y Black Forest Labs cuantizaron los modelos a FP8, reduciendo los requisitos de VRAM en un 40%, manteniendo una calidad de imagen comparable.
• Streaming de pesos mejorado: NVIDIA se asoció con ComfyUI para actualizar la función de “streaming de pesos”, permitiendo que modelos masivos se ejecuten en GPU GeForce RTX al transferir datos a la RAM del sistema cuando la memoria de la GPU sea limitada.

Cualquier persona puede comenzar a experimentar con estos nuevos modelos en sus GPU GeForce RTX. Para empezar, se debe actualizar ComfyUI para acceder a las plantillas FLUX.2 o visiten la página de Black Forest Labs en Hugging Face para descargar los pesos del modelo. 

Lean la publicación de esta semana en RTX AI Garage para más detalles sobre cómo configurar estas optimizaciones y maximizar el rendimiento en sus PC RTX.