Stability AI construiește modele de bază pe Amazon SageMaker

Suntem încântați să anunțăm asta Stabilitate AI a selectat AWS ca furnizor de cloud preferat pentru a-și alimenta modelele AI de ultimă generație pentru generarea de imagini, limbă, audio, video și conținut 3D. Stability AI este o companie de inteligență artificială (AI) deschisă, condusă de comunitate, care dezvoltă tehnologii inovatoare. Cu Amazon SageMaker, Stability AI va construi modele AI pe clustere de calcul cu mii de GPU sau AWS Trainium jetoane, reducând timpul și costul de antrenament cu 58%. Stability AI va colabora, de asemenea, cu AWS pentru a le permite studenților, cercetătorilor, startup-urilor și întreprinderilor din întreaga lume să-și folosească instrumentele și modelele open-source.

„Misiunea noastră la Stability AI este să construim fundația pentru a activa potențialul umanității prin AI. AWS a fost un partener esențial în scalarea modelelor noastre de fundație open-source în diferite modalități și suntem încântați să le aducem la SageMaker pentru a permite zeci de mii de dezvoltatori și milioane de utilizatori să profite de ele. Așteptăm cu nerăbdare să vedem lucrurile uimitoare construite pe aceste modele și să ne ajutăm clienții să personalizeze și să-și scaleze modelele și soluțiile.”

-Emad Mostaque, fondator și CEO al Stability AI.

Modele AI generative și difuzie stabilă

Modelele AI generative pot crea text, imagini, audio, video, cod și multe altele din instrucțiuni simple text. De exemplu, am creat următoarea imagine dând acest text prompt modelului: „Patru oameni mergând pe bicicletă în Alpii elvețieni, pictură renascentist, scenă epică a naturii uluitoare, lumină difuză.” Am folosit un caiet Jupyter Amazon SageMaker Studio pentru a genera această imagine cu Stable Diffusion.

Stability AI a anunțat, de asemenea, un model de difuzie stabilă distilat, care poate genera imagini coerente de până la zece ori mai rapid decât înainte. Acesta este cel mai recent open-source lansarea introduce, de asemenea, modele pentru a mări rezoluția unei imagini și pentru a deduce informații de profunzime pentru a genera noi imagini. Următoarele imagini arată un exemplu despre cum puteți utiliza noul model depth2img pentru a genera imagini noi, păstrând în același timp profunzimea și coerența imaginii originale.

Suntem încântați de potențialul acestor modele AI generative și de ceea ce vor crea clienții noștri. Din vopsirea la inversare textuală la modificatori, comunitatea continuă să inoveze și să construiască modele și instrumente open-source mai bune în IA generativă.

Antrenează modele de fundație la scară cu SageMaker

Modelele de bază — modele mari care sunt adaptabile la o varietate de sarcini din aval în domenii precum limbaj, imagine, audio, video — sunt greu de antrenat, deoarece necesită un cluster de calcul de înaltă performanță cu mii de cipuri GPU sau Trainium, împreună cu software. pentru a utiliza eficient clusterul.

Stability AI a ales AWS ca furnizor de cloud preferat pentru a furniza unul dintre cele mai mari clustere de GPU din cloudul public. Folosind infrastructura gestionată și bibliotecile de optimizare ale SageMaker, Stability este capabil să facă antrenamentul modelului său mai rezistent și mai performant. De exemplu, cu modele precum GPT NeoX, Stability AI a reușit să reducă timpul și costurile de antrenament cu 58% folosind SageMaker și bibliotecă paralelă de modele. Aceste optimizări și îmbunătățiri ale performanței se aplică modelelor cu zeci sau sute de miliarde de parametri.

Începeți cu Stable Diffusion

Stable Diffusion 2.0 este disponibil astăzi pe Amazon SageMaker JumpStart. JumpStart este centrul de învățare automată (ML) al SageMaker, care oferă sute de algoritmi încorporați, modele pre-antrenate și șabloane de soluții end-to-end pentru a vă ajuta să începeți rapid cu ML.

Începeți azi cu Difuziune stabilă 2.0.

Despre autori

Aditya Bindal este manager de produs principal pentru AWS Deep Learning. Lucrează la software și instrumente pentru a ușura clienților instruirea la scară largă și inferența. În timpul liber, îi place să petreacă timpul cu fiica sa, să joace tenis, să citească ficțiune istorică și să călătorească.