Stability AI bygger grundmodeller på Amazon SageMaker

Vi är glada över att kunna meddela det Stabilitet AI har valt AWS som sin föredragna molnleverantör för att driva sina toppmoderna AI-modeller för bild-, språk-, ljud-, video- och 3D-innehållsgenerering. Stability AI är ett gemenskapsdrivet företag med öppen källkod för artificiell intelligens (AI) som utvecklar banbrytande teknologier. Med Amazon SageMaker, Stabilitet AI kommer att bygga AI-modeller på datorkluster med tusentals GPU eller AWS Trainium marker, vilket minskar träningstiden och -kostnaderna med 58 %. Stability AI kommer också att samarbeta med AWS för att göra det möjligt för studenter, forskare, startups och företag runt om i världen att använda sina verktyg och modeller med öppen källkod.

"Vårt uppdrag på Stability AI är att bygga grunden för att aktivera mänsklighetens potential genom AI. AWS har varit en integrerad partner i att skala våra grundmodeller med öppen källkod över olika modaliteter, och vi är glada över att kunna ta dessa till SageMaker för att göra det möjligt för tiotusentals utvecklare och miljontals användare att dra nytta av dem. Vi ser fram emot att se de fantastiska sakerna som bygger på dessa modeller och hjälpa våra kunder att anpassa och skala sina modeller och lösningar.”

-Emad Mostaque, grundare och VD för Stability AI.

Generativa AI-modeller och stabil diffusion

Generativa AI-modeller kan skapa text, bilder, ljud, video, kod och mer från enkla textinstruktioner. Till exempel skapade jag följande bild genom att ge denna textuppmaning till modellen: "Fyra personer som cyklar i de schweiziska alperna, renässansmålning, episk hisnande naturscen, diffust ljus." Jag använde en Jupyter-anteckningsbok i Amazon SageMaker Studio för att generera den här bilden med stabil diffusion.

Stability AI tillkännagav också en destillerad stabil diffusionsmodell, som kan generera sammanhängande bilder upp till tio gånger snabbare än tidigare. Detta senaste öppen källkod release introducerar också modeller för att uppskala en bilds upplösning och härleda djupinformation för att generera nya bilder. Följande bilder visar ett exempel på hur du kan använda den nya depth2img-modellen för att generera nya bilder samtidigt som originalbildens djup och koherens bevaras.

Vi är glada över potentialen hos dessa generativa AI-modeller och över vad våra kunder kommer att skapa. Från målning till textinversion till modifieringsmedel, fortsätter samhället att förnya och bygga bättre modeller och verktyg med öppen källkod inom generativ AI.

Utbilda grundmodeller i stor skala med SageMaker

Grundmodeller – stora modeller som kan anpassas till en mängd olika nedströmsuppgifter inom domäner som språk, bild, ljud, video – är svåra att träna eftersom de kräver ett högpresterande datorkluster med tusentals GPU- eller Trainium-chips, tillsammans med programvara för att effektivt använda klustret.

Stabilitet AI valde AWS som sin föredragna molnleverantör för att tillhandahålla ett av de största klustren av GPU:er någonsin i det offentliga molnet. Genom att använda SageMakers hanterade infrastruktur och optimeringsbibliotek kan Stability göra sin modellträning mer motståndskraftig och presterande. Till exempel, med modeller som GPT NeoX kunde Stability AI minska träningstiden och -kostnaderna med 58 % med SageMaker och dess modell parallellbibliotek. Dessa optimeringar och prestandaförbättringar gäller modeller med tiotals eller hundratals miljarder parametrar.

Kom igång med Stable Diffusion

Stable Diffusion 2.0 är tillgänglig idag på Amazon SageMaker JumpStart. JumpStart är maskininlärningsnavet (ML) i SageMaker som tillhandahåller hundratals inbyggda algoritmer, förutbildade modeller och helhetslösningsmallar för att hjälpa dig att snabbt komma igång med ML.

Kom igång idag med Stabil diffusion 2.0.

Om författarna

Aditya Bindal är huvudproduktchef för AWS Deep Learning. Han arbetar med mjukvara och verktyg för att göra storskalig utbildning och slutsatser lättare för kunderna. På fritiden tycker han om att umgås med sin dotter, spela tennis, läsa historisk skönlitteratur och resa.