Stability AI crée des modèles de base sur Amazon SageMaker

Nous sommes ravis d'annoncer que IA de stabilité a choisi AWS comme fournisseur cloud préféré pour alimenter ses modèles d'IA de pointe pour la génération de contenu image, langage, audio, vidéo et 3D. Stability AI est une société d'intelligence artificielle (IA) open source axée sur la communauté qui développe des technologies révolutionnaires. Avec Amazon Sage Maker, Stability AI construira des modèles d'IA sur des clusters de calcul avec des milliers de GPU ou Formation AWS puces, réduisant le temps et les coûts de formation de 58 %. Stability AI collaborera également avec AWS pour permettre aux étudiants, chercheurs, startups et entreprises du monde entier d'utiliser ses outils et modèles open source.

"Notre mission chez Stability AI est de jeter les bases pour activer le potentiel de l'humanité grâce à l'IA. AWS a été un partenaire à part entière dans la mise à l'échelle de nos modèles de base open source à travers les modalités, et nous sommes ravis de les apporter à SageMaker pour permettre à des dizaines de milliers de développeurs et à des millions d'utilisateurs d'en tirer parti. Nous sommes impatients de voir les choses incroyables construites sur ces modèles et d'aider nos clients à personnaliser et à faire évoluer leurs modèles et leurs solutions.

-Emad Mostaque, fondateur et PDG de Stability AI.

Modèles d'IA générative et diffusion stable

Les modèles d'IA générative peuvent créer du texte, des images, de l'audio, de la vidéo, du code et plus encore à partir d'instructions textuelles simples. Par exemple, j'ai créé l'image suivante en donnant cette invite de texte au modèle : "Quatre personnes faisant du vélo dans les Alpes suisses, peinture de la Renaissance, scène de nature épique à couper le souffle, lumière diffuse." J'ai utilisé un cahier Jupyter dans Amazon SageMakerStudio pour générer cette image avec Stable Diffusion.

Stability AI a également annoncé un modèle de diffusion stable distillé, qui peut générer des images cohérentes jusqu'à dix fois plus rapidement qu'auparavant. Ce dernier open-source La version introduit également des modèles pour augmenter la résolution d'une image et déduire des informations de profondeur pour générer de nouvelles images. Les images suivantes montrent un exemple de la façon dont vous pouvez utiliser le nouveau modèle depth2img pour générer de nouvelles images tout en préservant la profondeur et la cohérence de l'image d'origine.

Nous sommes enthousiasmés par le potentiel de ces modèles d'IA générative et par ce que nos clients vont créer. De peinture à inversion textuelle à modificateurs, la communauté continue d'innover et de créer de meilleurs modèles et outils open source dans le domaine de l'IA générative.

Former des modèles de base à grande échelle avec SageMaker

Les modèles de base (grands modèles adaptables à une variété de tâches en aval dans des domaines tels que le langage, l'image, l'audio, la vidéo) sont difficiles à former car ils nécessitent un cluster de calcul hautes performances avec des milliers de puces GPU ou Trainium, ainsi que des logiciels pour utiliser efficacement le cluster.

Stability AI a choisi AWS comme fournisseur de cloud préféré pour fournir l'un des plus grands clusters de GPU jamais créés dans le cloud public. En utilisant l'infrastructure gérée et les bibliothèques d'optimisation de SageMaker, Stability est en mesure de rendre la formation de son modèle plus résiliente et performante. Par exemple, avec des modèles tels que GPT NeoX, Stability AI a pu réduire le temps et les coûts de formation de 58 % en utilisant SageMaker et son bibliothèque parallèle de modèles. Ces optimisations et améliorations de performances s'appliquent aux modèles comportant des dizaines ou des centaines de milliards de paramètres.

Démarrer avec Stable Diffusion

Stable Diffusion 2.0 est disponible aujourd'hui sur Amazon SageMaker JumpStart. JumpStart est le centre d'apprentissage automatique (ML) de SageMaker qui fournit des centaines d'algorithmes intégrés, des modèles pré-formés et des modèles de solutions de bout en bout pour vous aider à démarrer rapidement avec ML.

Commencez dès aujourd'hui avec Diffusion stable 2.0.

À propos des auteurs

Aditya Bindal est chef de produit principal pour AWS Deep Learning. Il travaille sur des logiciels et des outils pour faciliter la formation et l'inférence à grande échelle pour les clients. Dans ses temps libres, il aime passer du temps avec sa fille, jouer au tennis, lire des romans historiques et voyager.