Stability AI bygger grunnmodeller på Amazon SageMaker

Det gleder oss over å kunngjøre det Stabilitet AI har valgt AWS som sin foretrukne skyleverandør for å drive sine toppmoderne AI-modeller for generering av bilde, språk, lyd, video og 3D-innhold. Stability AI er et fellesskapsdrevet, åpen kildekode-selskap for kunstig intelligens (AI) som utvikler banebrytende teknologier. Med Amazon SageMaker, Stabilitet AI vil bygge AI-modeller på dataklynger med tusenvis av GPU eller AWS Trainium sjetonger, noe som reduserer treningstid og kostnad med 58 %. Stability AI vil også samarbeide med AWS for å gjøre det mulig for studenter, forskere, startups og bedrifter over hele verden å bruke åpen kildekodeverktøy og modeller.

«Vårt oppdrag hos Stability AI er å bygge grunnlaget for å aktivere menneskehetens potensial gjennom AI. AWS har vært en integrert partner i å skalere våre open source-fundamentmodeller på tvers av modaliteter, og vi er glade for å bringe disse til SageMaker for å gjøre det mulig for titusenvis av utviklere og millioner av brukere å dra nytte av dem. Vi ser frem til å se de fantastiske tingene bygget på disse modellene og hjelpe kundene våre med å tilpasse og skalere modellene og løsningene deres.»

-Emad Mostaque, grunnlegger og administrerende direktør for Stability AI.

Generative AI-modeller og stabil diffusjon

Generative AI-modeller kan lage tekst, bilder, lyd, video, kode og mer fra enkle tekstinstruksjoner. For eksempel laget jeg følgende bilde ved å gi denne tekstmeldingen til modellen: «Fire personer som sykler i de sveitsiske alper, renessansemaleri, episk fantastisk naturscene, diffust lys.» Jeg brukte en Jupyter notatbok i Amazon SageMaker Studio for å generere dette bildet med stabil diffusjon.

Stability AI annonserte også en destillert stabil diffusjonsmodell, som kan generere sammenhengende bilder opptil ti ganger raskere enn før. Dette siste åpen kildekode utgivelsen introduserer også modeller for å oppskalere et bildes oppløsning og utlede dybdeinformasjon for å generere nye bilder. De følgende bildene viser et eksempel på hvordan du kan bruke den nye depth2img-modellen til å generere nye bilder samtidig som dybden og sammenhengen til det originale bildet bevares.

Vi er begeistret over potensialet til disse generative AI-modellene og av hva kundene våre vil skape. Fra maling til tekstuell inversjon til modifikatorer, fortsetter fellesskapet å innovere og bygge bedre åpen kildekode-modeller og verktøy i generativ AI.

Trening av grunnmodeller i stor skala med SageMaker

Grunnmodeller – store modeller som kan tilpasses en rekke nedstrømsoppgaver innen domener som språk, bilde, lyd, video – er vanskelige å trene fordi de krever en høyytelses datamaskinklynge med tusenvis av GPU- eller Trainium-brikker, sammen med programvare å utnytte klyngen effektivt.

Stabilitet AI valgte AWS som sin foretrukne skyleverandør for å levere en av de største klyngene av GPUer i den offentlige skyen noensinne. Ved å bruke SageMakers administrerte infrastruktur og optimaliseringsbiblioteker, er Stability i stand til å gjøre modelltreningen mer spenstig og mer effektiv. For eksempel, med modeller som GPT NeoX, var Stability AI i stand til å redusere treningstiden og -kostnadene med 58 % ved å bruke SageMaker og dets modell parallellbibliotek. Disse optimaliseringene og ytelsesforbedringene gjelder for modeller med titalls eller hundrevis av milliarder av parametere.

Kom i gang med Stable Diffusion

Stable Diffusion 2.0 er tilgjengelig i dag på Amazon SageMaker JumpStart. JumpStart er maskinlæringssenteret (ML) til SageMaker som gir hundrevis av innebygde algoritmer, forhåndsopplærte modeller og ende-til-ende løsningsmaler for å hjelpe deg raskt å komme i gang med ML.

Kom i gang i dag med Stabil diffusjon 2.0.

Om forfatterne

Aditya Bindal er hovedproduktsjef for AWS Deep Learning. Han jobber med programvare og verktøy for å gjøre opplæring og slutninger i stor skala enklere for kundene. På fritiden liker han å tilbringe tid med datteren, spille tennis, lese historisk skjønnlitteratur og reise.