Stability AI crea modelos básicos en Amazon SageMaker

Estamos encantados de anunciar que Estabilidad IA ha seleccionado a AWS como su proveedor de nube preferido para potenciar sus modelos de inteligencia artificial de última generación para la generación de imágenes, idiomas, audio, video y contenido 3D. Stability AI es una empresa de inteligencia artificial (IA) de código abierto impulsada por la comunidad que desarrolla tecnologías innovadoras. Con Amazon SageMaker, Stability AI creará modelos de IA en clústeres informáticos con miles de GPU o tren de AWS chips, lo que reduce el tiempo y el costo de capacitación en un 58 %. Stability AI también colaborará con AWS para permitir que estudiantes, investigadores, nuevas empresas y empresas de todo el mundo utilicen sus herramientas y modelos de código abierto.

“Nuestra misión en Stability AI es sentar las bases para activar el potencial de la humanidad a través de la IA. AWS ha sido un socio integral en la ampliación de nuestros modelos básicos de código abierto en todas las modalidades, y estamos encantados de traerlos a SageMaker para permitir que decenas de miles de desarrolladores y millones de usuarios los aprovechen. Esperamos ver las increíbles cosas construidas en estos modelos y ayudar a nuestros clientes a personalizar y escalar sus modelos y soluciones”.

-Emad Mostaque, Fundador y CEO de Stability AI.

Modelos generativos de IA y difusión estable

Los modelos generativos de IA pueden crear texto, imágenes, audio, video, código y más a partir de instrucciones de texto simples. Por ejemplo, creé la siguiente imagen dando este mensaje de texto al modelo: "Cuatro personas en bicicleta en los Alpes suizos, pintura renacentista, escena natural épica e impresionante, luz difusa". Usé un cuaderno Jupyter en Estudio Amazon SageMaker para generar esta imagen con Stable Diffusion.

Stability AI también anunció un modelo de difusión estable destilado, que puede generar imágenes coherentes hasta diez veces más rápido que antes. este último De código abierto La versión también presenta modelos para aumentar la resolución de una imagen e inferir información de profundidad para generar nuevas imágenes. Las siguientes imágenes muestran un ejemplo de cómo puede utilizar el nuevo modelo depth2img para generar nuevas imágenes conservando la profundidad y la coherencia de la imagen original.

Estamos entusiasmados con el potencial de estos modelos generativos de IA y con lo que crearán nuestros clientes. De pintura a inversión textual a modificadores, la comunidad continúa innovando y construyendo mejores modelos y herramientas de código abierto en IA generativa.

Capacitación de modelos básicos a escala con SageMaker

Los modelos básicos (modelos grandes que se adaptan a una variedad de tareas posteriores en dominios como lenguaje, imagen, audio, video) son difíciles de entrenar porque requieren un clúster de cómputo de alto rendimiento con miles de GPU o chips Trainium, junto con software. para utilizar eficientemente el clúster.

Stability AI eligió a AWS como su proveedor de nube preferido para aprovisionar uno de los clústeres de GPU más grandes de la historia en la nube pública. Mediante el uso de la infraestructura administrada y las bibliotecas de optimización de SageMaker, Stability puede hacer que el entrenamiento de su modelo sea más resistente y eficaz. Por ejemplo, con modelos como GPT NeoX, Stability AI pudo reducir el tiempo y el costo de la capacitación en un 58 % con SageMaker y su biblioteca modelo paralelo. Estas optimizaciones y mejoras de rendimiento se aplican a modelos con decenas o cientos de miles de millones de parámetros.

Comience con la difusión estable

Stable Diffusion 2.0 está disponible hoy en JumpStart de Amazon SageMaker. JumpStart es el centro de aprendizaje automático (ML) de SageMaker que proporciona cientos de algoritmos integrados, modelos previamente entrenados y plantillas de soluciones integrales para ayudarlo a comenzar rápidamente con ML.

Comience hoy con Difusión estable 2.0.

Sobre los autores

Aditya Bindal es gerente principal de productos para AWS Deep Learning. Trabaja en software y herramientas para facilitar la formación y la inferencia a gran escala para los clientes. En su tiempo libre, disfruta pasar tiempo con su hija, jugar tenis, leer ficción histórica y viajar.