Stability AI bygger fundamentmodeller på Amazon SageMaker

Vi er glade for at kunne meddele det Stabilitet AI har valgt AWS som sin foretrukne cloud-udbyder til at drive sine avancerede AI-modeller til billed-, sprog-, lyd-, video- og 3D-indholdsgenerering. Stability AI er en fællesskabsdrevet, open source kunstig intelligens (AI) virksomhed, der udvikler banebrydende teknologier. Med Amazon SageMaker, Stabilitet AI vil bygge AI-modeller på computerklynger med tusindvis af GPU eller AWS Trainium chips, hvilket reducerer træningstiden og omkostningerne med 58 %. Stability AI vil også samarbejde med AWS for at gøre det muligt for studerende, forskere, startups og virksomheder rundt om i verden at bruge dets open source-værktøjer og -modeller.

"Vores mission hos Stability AI er at bygge fundamentet for at aktivere menneskehedens potentiale gennem AI. AWS har været en integreret partner i at skalere vores open source-fundamentmodeller på tværs af modaliteter, og vi er glade for at bringe disse til SageMaker for at gøre det muligt for titusindvis af udviklere og millioner af brugere at drage fordel af dem. Vi ser frem til at se de fantastiske ting bygget på disse modeller og hjælpe vores kunder med at tilpasse og skalere deres modeller og løsninger."

-Emad Mostaque, grundlægger og administrerende direktør for Stability AI.

Generative AI-modeller og stabil diffusion

Generative AI-modeller kan skabe tekst, billeder, lyd, video, kode og mere ud fra simple tekstinstruktioner. For eksempel oprettede jeg følgende billede ved at give denne tekstprompt til modellen: "Fire mennesker, der kører på cykel i de schweiziske alper, renæssancemaleri, episk betagende naturscene, diffust lys." Jeg brugte en Jupyter notesbog i Amazon SageMaker Studio for at generere dette billede med stabil diffusion.

Stabilitet AI annoncerede også en destilleret stabil diffusionsmodel, som kan generere sammenhængende billeder op til ti gange hurtigere end før. Dette seneste open source release introducerer også modeller til at opskalere et billedes opløsning og udlede dybdeinformation for at generere nye billeder. De følgende billeder viser et eksempel på, hvordan du kan bruge den nye depth2img-model til at generere nye billeder og samtidig bevare dybden og sammenhængen i det originale billede.

Vi er begejstrede over potentialet i disse generative AI-modeller og over, hvad vores kunder vil skabe. Fra indmaling til tekstinversion til modifikatorer, fortsætter fællesskabet med at innovere og bygge bedre open source-modeller og værktøjer i generativ AI.

Træning af fundamentmodeller i skala med SageMaker

Grundmodeller – store modeller, der kan tilpasses til en række downstream-opgaver inden for domæner som sprog, billede, lyd, video – er svære at træne, fordi de kræver en højtydende computerklynge med tusindvis af GPU- eller Trainium-chips sammen med software at udnytte klyngen effektivt.

Stabilitet AI valgte AWS som sin foretrukne cloud-udbyder til at levere en af ​​de største klynger af GPU'er nogensinde i den offentlige sky. Ved at bruge SageMakers administrerede infrastruktur og optimeringsbiblioteker er Stability i stand til at gøre sin modeltræning mere modstandsdygtig og mere effektiv. For eksempel, med modeller som GPT NeoX var Stability AI i stand til at reducere træningstiden og omkostningerne med 58 % ved at bruge SageMaker og dets model parallelbibliotek. Disse optimeringer og ydeevneforbedringer gælder for modeller med titusinder eller hundreder af milliarder af parametre.

Kom godt i gang med Stable Diffusion

Stable Diffusion 2.0 er tilgængelig i dag på Amazon SageMaker JumpStart. JumpStart er maskinlæringscentret (ML) i SageMaker, der giver hundredvis af indbyggede algoritmer, forudtrænede modeller og end-to-end løsningsskabeloner, der hjælper dig hurtigt med at komme i gang med ML.

Kom i gang i dag med Stabil diffusion 2.0.

Om forfatterne

Aditya Bindal er hovedproduktchef for AWS Deep Learning. Han arbejder på software og værktøjer for at gøre omfattende uddannelse og konklusioner lettere for kunderne. I sin fritid nyder han at bruge tid sammen med sin datter, spille tennis, læse historisk skønlitteratur og rejse.