Stability AI cria modelos de base no Amazon SageMaker

Temos o prazer de anunciar que IA de estabilidade selecionou a AWS como seu provedor de nuvem preferido para capacitar seus modelos de IA de última geração para geração de imagens, idiomas, áudio, vídeo e conteúdo 3D. A Stability AI é uma empresa de inteligência artificial (IA) de código aberto voltada para a comunidade que desenvolve tecnologias inovadoras. Com Amazon Sage Maker, Stability AI construirá modelos de AI em clusters de computação com milhares de GPU ou Treinamento AWS chips, reduzindo tempo e custo de treinamento em 58%. A Stability AI também colaborará com a AWS para permitir que estudantes, pesquisadores, startups e empresas de todo o mundo usem suas ferramentas e modelos de código aberto.

“Nossa missão na Stability AI é construir a base para ativar o potencial da humanidade por meio da IA. A AWS tem sido um parceiro integral no dimensionamento de nossos modelos de base de código aberto em várias modalidades, e estamos muito satisfeitos em trazê-los para o SageMaker para permitir que dezenas de milhares de desenvolvedores e milhões de usuários aproveitem-nos. Estamos ansiosos para ver as coisas incríveis construídas nesses modelos e ajudar nossos clientes a personalizar e dimensionar seus modelos e soluções”.

-Emad Mostaque, fundador e CEO da Stability AI.

Modelos generativos de IA e difusão estável

Os modelos generativos de IA podem criar texto, imagens, áudio, vídeo, código e muito mais a partir de instruções de texto simples. Por exemplo, criei a seguinte imagem dando este prompt de texto ao modelo: “Quatro pessoas andando de bicicleta nos Alpes suíços, pintura renascentista, cena épica da natureza de tirar o fôlego, luz difusa”. Eu usei um notebook Jupyter em Estúdio Amazon SageMaker para gerar esta imagem com difusão estável.

Stability AI também anunciou um modelo de difusão estável destilado, que pode gerar imagens coerentes até dez vezes mais rápido do que antes. este último de código aberto O lançamento também apresenta modelos para aumentar a resolução de uma imagem e inferir informações de profundidade para gerar novas imagens. As imagens a seguir mostram um exemplo de como você pode usar o novo modelo depth2img para gerar novas imagens preservando a profundidade e a coerência da imagem original.

Estamos entusiasmados com o potencial desses modelos generativos de IA e com o que nossos clientes criarão. A partir de pintura para inversão textual para modificadores, a comunidade continua inovando e construindo melhores modelos e ferramentas de código aberto em IA generativa.

Modelos de base de treinamento em escala com o SageMaker

Modelos básicos — modelos grandes que são adaptáveis ​​a uma variedade de tarefas downstream em domínios como linguagem, imagem, áudio, vídeo — são difíceis de treinar porque exigem um cluster de computação de alto desempenho com milhares de chips GPU ou Trainium, juntamente com software para utilizar eficientemente o cluster.

A Stability AI escolheu a AWS como seu provedor de nuvem preferido para provisionar um dos maiores clusters de GPUs de todos os tempos na nuvem pública. Usando a infraestrutura gerenciada e as bibliotecas de otimização do SageMaker, o Stability pode tornar seu treinamento de modelo mais resiliente e eficiente. Por exemplo, com modelos como o GPT NeoX, a Stability AI conseguiu reduzir o tempo e o custo do treinamento em 58% usando o SageMaker e seu biblioteca paralela de modelos. Essas otimizações e melhorias de desempenho se aplicam a modelos com dezenas ou centenas de bilhões de parâmetros.

Comece com a difusão estável

Stable Diffusion 2.0 está disponível hoje em JumpStart do Amazon SageMaker. O JumpStart é o hub de aprendizado de máquina (ML) do SageMaker que fornece centenas de algoritmos integrados, modelos pré-treinados e modelos de solução de ponta a ponta para ajudá-lo a começar a usar o ML rapidamente.

Comece hoje mesmo com Difusão estável 2.0.

Sobre os autores

Aditya Bindal é gerente de produto principal da AWS Deep Learning. Ele trabalha com software e ferramentas para facilitar o treinamento e a inferência em grande escala para os clientes. Em seu tempo livre, ele gosta de passar o tempo com sua filha, jogando tênis, lendo ficção histórica e viajando.