Buckets:

rtrm's picture
download
raw
36.1 kB
import{s as Fl,n as Yl,o as Ql}from"../chunks/scheduler.94020406.js";import{S as Vl,i as zl,g as n,s as a,r as m,E as Rl,h as i,f as l,c as p,j as Xl,u as M,x as r,k as Hl,y as Nl,a as s,v as f,d as o,t as y,w}from"../chunks/index.a08c8d92.js";import{C as b}from"../chunks/CodeBlock.b23cf525.js";import{D as Sl}from"../chunks/DocNotebookDropdown.d8a25975.js";import{H as J,E as Ll}from"../chunks/EditOnGithub.b1bceb47.js";function Al(tl){let u,Xe,Ze,He,j,Fe,$,Ye,c,ll='πŸ€— Hugging Face [Diffusers] (<a href="https://github.com/huggingface/diffusers" rel="nofollow">https://github.com/huggingface/diffusers</a>) λŠ” 버전 0.5.1λΆ€ν„° Flaxλ₯Ό μ§€μ›ν•©λ‹ˆλ‹€! 이λ₯Ό 톡해 Colab, Kaggle, Google Cloud Platformμ—μ„œ μ‚¬μš©ν•  수 μžˆλŠ” κ²ƒμ²˜λŸΌ Google TPUμ—μ„œ μ΄ˆκ³ μ† 좔둠이 κ°€λŠ₯ν•©λ‹ˆλ‹€.',Qe,T,sl='이 λ…ΈνŠΈλΆμ€ JAX / Flaxλ₯Ό μ‚¬μš©ν•΄ 좔둠을 μ‹€ν–‰ν•˜λŠ” 방법을 λ³΄μ—¬μ€λ‹ˆλ‹€. Stable Diffusion의 μž‘λ™ 방식에 λŒ€ν•œ μžμ„Έν•œ λ‚΄μš©μ„ μ›ν•˜κ±°λ‚˜ GPUμ—μ„œ μ‹€ν–‰ν•˜λ €λ©΄ 이 [λ…ΈνŠΈλΆ] ](<a href="https://huggingface.co/docs/diffusers/stable_diffusion)%EC%9D%84" rel="nofollow">https://huggingface.co/docs/diffusers/stable_diffusion)을</a> μ°Έμ‘°ν•˜μ„Έμš”.',Ve,d,al="λ¨Όμ €, TPU λ°±μ—”λ“œλ₯Ό μ‚¬μš©ν•˜κ³  μžˆλŠ”μ§€ ν™•μΈν•©λ‹ˆλ‹€. Colabμ—μ„œ 이 λ…ΈνŠΈλΆμ„ μ‹€ν–‰ν•˜λŠ” 경우, λ©”λ‰΄μ—μ„œ λŸ°νƒ€μž„μ„ μ„ νƒν•œ λ‹€μŒ β€œλŸ°νƒ€μž„ μœ ν˜• 변경” μ˜΅μ…˜μ„ μ„ νƒν•œ λ‹€μŒ ν•˜λ“œμ›¨μ–΄ 가속기 μ„€μ •μ—μ„œ TPUλ₯Ό μ„ νƒν•©λ‹ˆλ‹€.",ze,h,pl="JAXλŠ” TPU μ „μš©μ€ μ•„λ‹ˆμ§€λ§Œ 각 TPU μ„œλ²„μ—λŠ” 8개의 TPU 가속기가 λ³‘λ ¬λ‘œ μž‘λ™ν•˜κΈ° λ•Œλ¬Έμ— ν•΄λ‹Ή ν•˜λ“œμ›¨μ–΄μ—μ„œ 더 빛을 λ°œν•œλ‹€λŠ” 점은 μ•Œμ•„λ‘μ„Έμš”.",Re,U,Ne,g,nl="λ¨Όμ € diffusersκ°€ μ„€μΉ˜λ˜μ–΄ μžˆλŠ”μ§€ ν™•μΈν•©λ‹ˆλ‹€.",Se,B,Le,C,Ae,_,Pe,I,Ee,k,il="그런 λ‹€μŒ λͺ¨λ“  dependenciesλ₯Ό κ°€μ Έμ˜΅λ‹ˆλ‹€.",qe,x,De,v,Ke,G,ml="TPU μž₯μΉ˜λŠ” 효율적인 half-float μœ ν˜•μΈ bfloat16을 μ§€μ›ν•©λ‹ˆλ‹€. ν…ŒμŠ€νŠΈμ—λŠ” 이 μœ ν˜•μ„ μ‚¬μš©ν•˜μ§€λ§Œ λŒ€μ‹  float32λ₯Ό μ‚¬μš©ν•˜μ—¬ 전체 정밀도(full precision)λ₯Ό μ‚¬μš©ν•  μˆ˜λ„ μžˆμŠ΅λ‹ˆλ‹€.",Oe,Z,et,W,Ml="FlaxλŠ” ν•¨μˆ˜ν˜• ν”„λ ˆμž„μ›Œν¬μ΄λ―€λ‘œ λͺ¨λΈμ€ λ¬΄μƒνƒœ(stateless)ν˜•μ΄λ©° λ§€κ°œλ³€μˆ˜λŠ” λͺ¨λΈ 외뢀에 μ €μž₯λ©λ‹ˆλ‹€. μ‚¬μ „ν•™μŠ΅λœ Flax νŒŒμ΄ν”„λΌμΈμ„ 뢈러였면 νŒŒμ΄ν”„λΌμΈ μžμ²΄μ™€ λͺ¨λΈ κ°€μ€‘μΉ˜(λ˜λŠ” λ§€κ°œλ³€μˆ˜)κ°€ λͺ¨λ‘ λ°˜ν™˜λ©λ‹ˆλ‹€. μ €ν¬λŠ” bf16 λ²„μ „μ˜ κ°€μ€‘μΉ˜λ₯Ό μ‚¬μš©ν•˜κ³  μžˆμœΌλ―€λ‘œ μœ ν˜• κ²½κ³ κ°€ ν‘œμ‹œλ˜μ§€λ§Œ λ¬΄μ‹œν•΄λ„ λ©λ‹ˆλ‹€.",tt,X,lt,H,st,F,fl="TPUμ—λŠ” 일반적으둜 8개의 λ””λ°”μ΄μŠ€κ°€ λ³‘λ ¬λ‘œ μž‘λ™ν•˜λ―€λ‘œ λ³΄μœ ν•œ λ””λ°”μ΄μŠ€ 수만큼 ν”„λ‘¬ν”„νŠΈλ₯Ό λ³΅μ œν•©λ‹ˆλ‹€. 그런 λ‹€μŒ 각각 ν•˜λ‚˜μ˜ 이미지 생성을 λ‹΄λ‹Ήν•˜λŠ” 8개의 λ””λ°”μ΄μŠ€μ—μ„œ ν•œ λ²ˆμ— 좔둠을 μˆ˜ν–‰ν•©λ‹ˆλ‹€. λ”°λΌμ„œ ν•˜λ‚˜μ˜ 칩이 ν•˜λ‚˜μ˜ 이미지λ₯Ό μƒμ„±ν•˜λŠ” 데 κ±Έλ¦¬λŠ” μ‹œκ°„κ³Ό λ™μΌν•œ μ‹œκ°„μ— 8개의 이미지λ₯Ό 얻을 수 μžˆμŠ΅λ‹ˆλ‹€.",at,Y,ol="ν”„λ‘¬ν”„νŠΈλ₯Ό λ³΅μ œν•˜κ³  λ‚˜λ©΄ νŒŒμ΄ν”„λΌμΈμ˜ <code>prepare_inputs</code> ν•¨μˆ˜λ₯Ό ν˜ΈμΆœν•˜μ—¬ ν† ν°ν™”λœ ν…μŠ€νŠΈ IDλ₯Ό μ–»μŠ΅λ‹ˆλ‹€. ν† ν°ν™”λœ ν…μŠ€νŠΈμ˜ κΈΈμ΄λŠ” κΈ°λ³Έ CLIP ν…μŠ€νŠΈ λͺ¨λΈμ˜ ꡬ성에 따라 77ν† ν°μœΌλ‘œ μ„€μ •λ©λ‹ˆλ‹€.",pt,Q,nt,V,it,z,mt,R,yl="λͺ¨λΈ λ§€κ°œλ³€μˆ˜μ™€ μž…λ ₯값은 μš°λ¦¬κ°€ λ³΄μœ ν•œ 8개의 병렬 μž₯μΉ˜μ— 볡사(Replication)λ˜μ–΄μ•Ό ν•©λ‹ˆλ‹€. λ§€κ°œλ³€μˆ˜ λ”•μ…”λ„ˆλ¦¬λŠ” <code>flax.jax_utils.replicate</code>(λ”•μ…”λ„ˆλ¦¬λ₯Ό μˆœνšŒν•˜λ©° κ°€μ€‘μΉ˜μ˜ λͺ¨μ–‘을 λ³€κ²½ν•˜μ—¬ 8번 λ°˜λ³΅ν•˜λŠ” ν•¨μˆ˜)λ₯Ό μ‚¬μš©ν•˜μ—¬ λ³΅μ‚¬λ©λ‹ˆλ‹€. 배열은 <code>shard</code>λ₯Ό μ‚¬μš©ν•˜μ—¬ λ³΅μ œλ©λ‹ˆλ‹€.",Mt,N,ft,S,ot,L,yt,A,wl="이 shape은 8개의 λ””λ°”μ΄μŠ€ 각각이 shape <code>(1, 77)</code>의 jnp 배열을 μž…λ ₯κ°’μœΌλ‘œ λ°›λŠ”λ‹€λŠ” μ˜λ―Έμž…λ‹ˆλ‹€. 즉 1은 λ””λ°”μ΄μŠ€λ‹Ή batch(배치) ν¬κΈ°μž…λ‹ˆλ‹€. λ©”λͺ¨λ¦¬κ°€ μΆ©λΆ„ν•œ TPUμ—μ„œλŠ” ν•œ λ²ˆμ— μ—¬λŸ¬ 이미지(μΉ©λ‹Ή)λ₯Ό μƒμ„±ν•˜λ €λŠ” 경우 1보닀 클 수 μžˆμŠ΅λ‹ˆλ‹€.",wt,P,rl="이미지λ₯Ό 생성할 μ€€λΉ„κ°€ 거의 μ™„λ£Œλ˜μ—ˆμŠ΅λ‹ˆλ‹€! 이제 생성 ν•¨μˆ˜μ— 전달할 λ‚œμˆ˜ μƒμ„±κΈ°λ§Œ λ§Œλ“€λ©΄ λ©λ‹ˆλ‹€. 이것은 λ‚œμˆ˜λ₯Ό λ‹€λ£¨λŠ” λͺ¨λ“  ν•¨μˆ˜μ— λ‚œμˆ˜ 생성기가 μžˆμ–΄μ•Ό ν•œλ‹€λŠ”, λ‚œμˆ˜μ— λŒ€ν•΄ 맀우 μ§„μ§€ν•˜κ³  독단적인 Flax의 ν‘œμ€€ μ ˆμ°¨μž…λ‹ˆλ‹€. μ΄λ ‡κ²Œ ν•˜λ©΄ μ—¬λŸ¬ λΆ„μ‚°λœ κΈ°κΈ°μ—μ„œ ν›ˆλ ¨ν•  λ•Œμ—λ„ μž¬ν˜„μ„±μ΄ 보μž₯λ©λ‹ˆλ‹€.",rt,E,bl="μ•„λž˜ 헬퍼 ν•¨μˆ˜λŠ” μ‹œλ“œλ₯Ό μ‚¬μš©ν•˜μ—¬ λ‚œμˆ˜ 생성기λ₯Ό μ΄ˆκΈ°ν™”ν•©λ‹ˆλ‹€. λ™μΌν•œ μ‹œλ“œλ₯Ό μ‚¬μš©ν•˜λŠ” ν•œ μ •ν™•νžˆ λ™μΌν•œ κ²°κ³Όλ₯Ό 얻을 수 μžˆμŠ΅λ‹ˆλ‹€. λ‚˜μ€‘μ— λ…ΈνŠΈλΆμ—μ„œ κ²°κ³Όλ₯Ό 탐색할 λ•Œμ—” λ‹€λ₯Έ μ‹œλ“œλ₯Ό 자유둭게 μ‚¬μš©ν•˜μ„Έμš”.",bt,q,ut,D,ul="rngλ₯Ό 얻은 λ‹€μŒ 8번 β€˜λΆ„ν• β€™ν•˜μ—¬ 각 λ””λ°”μ΄μŠ€κ°€ λ‹€λ₯Έ μ œλ„ˆλ ˆμ΄ν„°λ₯Ό μˆ˜μ‹ ν•˜λ„λ‘ ν•©λ‹ˆλ‹€. λ”°λΌμ„œ 각 λ””λ°”μ΄μŠ€λ§ˆλ‹€ λ‹€λ₯Έ 이미지가 μƒμ„±λ˜λ©° 전체 ν”„λ‘œμ„ΈμŠ€λ₯Ό μž¬ν˜„ν•  수 μžˆμŠ΅λ‹ˆλ‹€.",Jt,K,jt,O,Jl="JAX μ½”λ“œλŠ” 맀우 λΉ λ₯΄κ²Œ μ‹€ν–‰λ˜λŠ” 효율적인 ν‘œν˜„μœΌλ‘œ μ»΄νŒŒμΌν•  수 μžˆμŠ΅λ‹ˆλ‹€. ν•˜μ§€λ§Œ 후속 ν˜ΈμΆœμ—μ„œ λͺ¨λ“  μž…λ ₯이 λ™μΌν•œ λͺ¨μ–‘을 갖도둝 ν•΄μ•Ό ν•˜λ©°, κ·Έλ ‡μ§€ μ•ŠμœΌλ©΄ JAXκ°€ μ½”λ“œλ₯Ό λ‹€μ‹œ μ»΄νŒŒμΌν•΄μ•Ό ν•˜λ―€λ‘œ μ΅œμ ν™”λœ 속도λ₯Ό ν™œμš©ν•  수 μ—†μŠ΅λ‹ˆλ‹€.",$t,ee,jl="<code>jit = True</code>λ₯Ό 인수둜 μ „λ‹¬ν•˜λ©΄ Flax νŒŒμ΄ν”„λΌμΈμ΄ μ½”λ“œλ₯Ό μ»΄νŒŒμΌν•  수 μžˆμŠ΅λ‹ˆλ‹€. λ˜ν•œ λͺ¨λΈμ΄ μ‚¬μš© κ°€λŠ₯ν•œ 8개의 λ””λ°”μ΄μŠ€μ—μ„œ λ³‘λ ¬λ‘œ μ‹€ν–‰λ˜λ„λ‘ 보μž₯ν•©λ‹ˆλ‹€.",ct,te,$l="λ‹€μŒ 셀을 처음 μ‹€ν–‰ν•˜λ©΄ μ»΄νŒŒμΌν•˜λŠ” 데 μ‹œκ°„μ΄ 였래 κ±Έλ¦¬μ§€λ§Œ 이후 호좜(μž…λ ₯이 λ‹€λ₯Έ κ²½μš°μ—λ„)은 훨씬 λΉ¨λΌμ§‘λ‹ˆλ‹€. 예λ₯Ό λ“€μ–΄, ν…ŒμŠ€νŠΈν–ˆμ„ λ•Œ TPU v2-8μ—μ„œ μ»΄νŒŒμΌν•˜λŠ” 데 1λΆ„ 이상 κ±Έλ¦¬μ§€λ§Œ 이후 μΆ”λ‘  μ‹€ν–‰μ—λŠ” μ•½ 7μ΄ˆκ°€ κ±Έλ¦½λ‹ˆλ‹€.",Tt,le,dt,se,ht,ae,cl="λ°˜ν™˜λœ λ°°μ—΄μ˜ shape은 <code>(8, 1, 512, 512, 3)</code>μž…λ‹ˆλ‹€. 이λ₯Ό μž¬κ΅¬μ„±ν•˜μ—¬ 두 번째 차원을 μ œκ±°ν•˜κ³  512 Γ— 512 Γ— 3의 이미지 8개λ₯Ό 얻은 λ‹€μŒ PIL둜 λ³€ν™˜ν•©λ‹ˆλ‹€.",Ut,pe,gt,ne,Bt,ie,Tl="이미지λ₯Ό κ·Έλ¦¬λ“œμ— ν‘œμ‹œν•˜λŠ” λ„μš°λ―Έ ν•¨μˆ˜λ₯Ό λ§Œλ“€μ–΄ λ³΄κ² μŠ΅λ‹ˆλ‹€.",Ct,me,_t,Me,It,fe,dl='<img src="https://huggingface.co/datasets/YiYiXu/test-doc-assets/resolve/main/stable_diffusion_jax_how_to_cell_38_output_0.jpeg" alt="img"/>',kt,oe,xt,ye,hl="λͺ¨λ“  λ””λ°”μ΄μŠ€μ—μ„œ λ™μΌν•œ ν”„λ‘¬ν”„νŠΈλ₯Ό λ³΅μ œν•  ν•„μš”λŠ” μ—†μŠ΅λ‹ˆλ‹€. ν”„λ‘¬ν”„νŠΈ 2개λ₯Ό 각각 4λ²ˆμ”© μƒμ„±ν•˜κ±°λ‚˜ ν•œ λ²ˆμ— 8개의 μ„œλ‘œ λ‹€λ₯Έ ν”„λ‘¬ν”„νŠΈλ₯Ό μƒμ„±ν•˜λŠ” λ“± μ›ν•˜λŠ” 것은 무엇이든 ν•  수 μžˆμŠ΅λ‹ˆλ‹€. ν•œλ²ˆ ν•΄λ³΄μ„Έμš”!",vt,we,Ul="λ¨Όμ € μž…λ ₯ μ€€λΉ„ μ½”λ“œλ₯Ό νŽΈλ¦¬ν•œ ν•¨μˆ˜λ‘œ λ¦¬νŒ©ν„°λ§ν•˜κ² μŠ΅λ‹ˆλ‹€:",Gt,re,Zt,be,Wt,ue,gl='<img src="https://huggingface.co/datasets/YiYiXu/test-doc-assets/resolve/main/stable_diffusion_jax_how_to_cell_43_output_0.jpeg" alt="img"/>',Xt,Je,Ht,je,Bl="μ•žμ„œ <code>diffusers</code> Flax νŒŒμ΄ν”„λΌμΈμ΄ λͺ¨λΈμ„ μžλ™μœΌλ‘œ μ»΄νŒŒμΌν•˜κ³  μ‚¬μš© κ°€λŠ₯ν•œ λͺ¨λ“  κΈ°κΈ°μ—μ„œ λ³‘λ ¬λ‘œ μ‹€ν–‰ν•œλ‹€κ³  λ§μ”€λ“œλ ΈμŠ΅λ‹ˆλ‹€. 이제 κ·Έ ν”„λ‘œμ„ΈμŠ€λ₯Ό κ°„λž΅ν•˜κ²Œ μ‚΄νŽ΄λ³΄κ³  μž‘λ™ 방식을 λ³΄μ—¬λ“œλ¦¬κ² μŠ΅λ‹ˆλ‹€.",Ft,$e,Cl='JAX λ³‘λ ¬ν™”λŠ” μ—¬λŸ¬ κ°€μ§€ λ°©λ²•μœΌλ‘œ μˆ˜ν–‰ν•  수 μžˆμŠ΅λ‹ˆλ‹€. κ°€μž₯ μ‰¬μš΄ 방법은 jax.pmap ν•¨μˆ˜λ₯Ό μ‚¬μš©ν•˜μ—¬ 단일 ν”„λ‘œκ·Έλž¨, 닀쀑 데이터(SPMD) 병렬화λ₯Ό λ‹¬μ„±ν•˜λŠ” κ²ƒμž…λ‹ˆλ‹€. 즉, λ™μΌν•œ μ½”λ“œμ˜ 볡사본을 각각 λ‹€λ₯Έ 데이터 μž…λ ₯에 λŒ€ν•΄ μ—¬λŸ¬ 개 μ‹€ν–‰ν•˜λŠ” κ²ƒμž…λ‹ˆλ‹€. 더 μ •κ΅ν•œ μ ‘κ·Ό 방식도 κ°€λŠ₯ν•˜λ―€λ‘œ 관심이 μžˆμœΌμ‹œλ‹€λ©΄ <a href="https://jax.readthedocs.io/en/latest/index.html" rel="nofollow">JAX λ¬Έμ„œ</a>와 <a href="https://jax.readthedocs.io/en/latest/jax-101/08-pjit.html?highlight=pjit" rel="nofollow"><code>pjit</code> νŽ˜μ΄μ§€</a>μ—μ„œ 이 주제λ₯Ό μ‚΄νŽ΄λ³΄μ‹œκΈ° λ°”λžλ‹ˆλ‹€!',Yt,ce,_l="<code>jax.pmap</code>은 두 κ°€μ§€ κΈ°λŠ₯을 μˆ˜ν–‰ν•©λ‹ˆλ‹€:",Qt,Te,Il="<li><code>jax.jit()</code>λ₯Ό ν˜ΈμΆœν•œ κ²ƒμ²˜λŸΌ μ½”λ“œλ₯Ό 컴파일(λ˜λŠ” <code>jit</code>)ν•©λ‹ˆλ‹€. 이 μž‘μ—…μ€ <code>pmap</code>을 ν˜ΈμΆœν•  λ•Œκ°€ μ•„λ‹ˆλΌ pmapped ν•¨μˆ˜κ°€ 처음 호좜될 λ•Œ μˆ˜ν–‰λ©λ‹ˆλ‹€.</li> <li>컴파일된 μ½”λ“œκ°€ μ‚¬μš© κ°€λŠ₯ν•œ λͺ¨λ“  κΈ°κΈ°μ—μ„œ λ³‘λ ¬λ‘œ μ‹€ν–‰λ˜λ„λ‘ ν•©λ‹ˆλ‹€.</li>",Vt,de,kl="μž‘λ™ 방식을 λ³΄μ—¬λ“œλ¦¬κΈ° μœ„ν•΄ 이미지 생성을 μ‹€ν–‰ν•˜λŠ” λΉ„κ³΅κ°œ λ©”μ„œλ“œμΈ νŒŒμ΄ν”„λΌμΈμ˜ <code>_generate</code> λ©”μ„œλ“œλ₯Ό <code>pmap</code>ν•©λ‹ˆλ‹€. 이 λ©”μ„œλ“œλŠ” ν–₯ν›„ <code>Diffusers</code> λ¦΄λ¦¬μŠ€μ—μ„œ 이름이 λ³€κ²½λ˜κ±°λ‚˜ 제거될 수 μžˆλ‹€λŠ” 점에 μœ μ˜ν•˜μ„Έμš”.",zt,he,Rt,Ue,xl="<code>pmap</code>을 μ‚¬μš©ν•œ ν›„ μ€€λΉ„λœ ν•¨μˆ˜ <code>p_generate</code>λŠ” κ°œλ…μ μœΌλ‘œ λ‹€μŒμ„ μˆ˜ν–‰ν•©λ‹ˆλ‹€:",Nt,ge,vl="<li>각 μž₯μΉ˜μ—μ„œ κΈ°λ³Έ ν•¨μˆ˜ <code>pipeline._generate</code>의 볡사본을 ν˜ΈμΆœν•©λ‹ˆλ‹€.</li> <li>각 μž₯μΉ˜μ— μž…λ ₯ 인수의 λ‹€λ₯Έ 뢀뢄을 λ³΄λƒ…λ‹ˆλ‹€. 이것이 λ°”λ‘œ 샀딩이 μ‚¬μš©λ˜λŠ” μ΄μœ μž…λ‹ˆλ‹€. 이 경우 <code>prompt_ids</code>의 shape은 <code>(8, 1, 77, 768)</code>μž…λ‹ˆλ‹€. 이 배열은 8개둜 λΆ„ν• λ˜κ³  <code>_generate</code>의 각 볡사본은 <code>(1, 77, 768)</code>의 shape을 κ°€μ§„ μž…λ ₯을 λ°›κ²Œ λ©λ‹ˆλ‹€.</li>",St,Be,Gl="λ³‘λ ¬λ‘œ ν˜ΈμΆœλœλ‹€λŠ” 사싀을 μ™„μ „νžˆ λ¬΄μ‹œν•˜κ³  <code>_generate</code>λ₯Ό μ½”λ”©ν•  수 μžˆμŠ΅λ‹ˆλ‹€. batch(배치) 크기(이 μ˜ˆμ œμ—μ„œλŠ” <code>1</code>)와 μ½”λ“œμ— μ ν•©ν•œ μ°¨μ›λ§Œ μ‹ κ²½ μ“°λ©΄ 되며, λ³‘λ ¬λ‘œ μž‘λ™ν•˜κΈ° μœ„ν•΄ 아무것도 λ³€κ²½ν•  ν•„μš”κ°€ μ—†μŠ΅λ‹ˆλ‹€.",Lt,Ce,Zl="νŒŒμ΄ν”„λΌμΈ ν˜ΈμΆœμ„ μ‚¬μš©ν•  λ•Œμ™€ λ§ˆμ°¬κ°€μ§€λ‘œ, λ‹€μŒ 셀을 처음 μ‹€ν–‰ν•  λ•ŒλŠ” μ‹œκ°„μ΄ κ±Έλ¦¬μ§€λ§Œ κ·Έ μ΄ν›„μ—λŠ” 훨씬 λΉ¨λΌμ§‘λ‹ˆλ‹€.",At,_e,Pt,Ie,Et,ke,qt,xe,Dt,ve,Wl="JAXλŠ” 비동기 λ””μŠ€νŒ¨μΉ˜λ₯Ό μ‚¬μš©ν•˜κ³  κ°€λŠ₯ν•œ ν•œ 빨리 μ œμ–΄κΆŒμ„ Python 루프에 λ°˜ν™˜ν•˜κΈ° λ•Œλ¬Έμ— μΆ”λ‘  μ‹œκ°„μ„ μ •ν™•ν•˜κ²Œ μΈ‘μ •ν•˜κΈ° μœ„ν•΄ <code>block_until_ready()</code>λ₯Ό μ‚¬μš©ν•©λ‹ˆλ‹€. 아직 κ΅¬μ²΄ν™”λ˜μ§€ μ•Šμ€ 계산 κ²°κ³Όλ₯Ό μ‚¬μš©ν•˜λ €λŠ” 경우 μžλ™μœΌλ‘œ 차단이 μˆ˜ν–‰λ˜λ―€λ‘œ μ½”λ“œμ—μ„œ 이 ν•¨μˆ˜λ₯Ό μ‚¬μš©ν•  ν•„μš”κ°€ μ—†μŠ΅λ‹ˆλ‹€.",Kt,Ge,Ot,We,el;return j=new J({props:{title:"JAX / Flaxμ—μ„œμ˜ 🧨 Stable Diffusion!",local:"jax--flaxμ—μ„œμ˜--stable-diffusion",headingTag:"h1"}}),$=new Sl({props:{classNames:"absolute z-10 right-0 top-0",options:[{label:"Mixed",value:"https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers_doc/ko/stable_diffusion_jax_how_to.ipynb"},{label:"PyTorch",value:"https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers_doc/ko/pytorch/stable_diffusion_jax_how_to.ipynb"},{label:"TensorFlow",value:"https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers_doc/ko/tensorflow/stable_diffusion_jax_how_to.ipynb"},{label:"Mixed",value:"https://studiolab.sagemaker.aws/import/github/huggingface/notebooks/blob/main/diffusers_doc/ko/stable_diffusion_jax_how_to.ipynb"},{label:"PyTorch",value:"https://studiolab.sagemaker.aws/import/github/huggingface/notebooks/blob/main/diffusers_doc/ko/pytorch/stable_diffusion_jax_how_to.ipynb"},{label:"TensorFlow",value:"https://studiolab.sagemaker.aws/import/github/huggingface/notebooks/blob/main/diffusers_doc/ko/tensorflow/stable_diffusion_jax_how_to.ipynb"}]}}),U=new J({props:{title:"Setup",local:"setup",headingTag:"h2"}}),B=new b({props:{code:"IXBpcCUyMGluc3RhbGwlMjBqYXglM0QlM0QwLjMuMjUlMjBqYXhsaWIlM0QlM0QwLjMuMjUlMjBmbGF4JTIwdHJhbnNmb3JtZXJzJTIwZnRmeSUwQSFwaXAlMjBpbnN0YWxsJTIwZGlmZnVzZXJz",highlighted:`!pip install jax==0.3.25 jaxlib==0.3.25 flax transformers ftfy
!pip install diffusers`,wrap:!1}}),C=new b({props:{code:"aW1wb3J0JTIwamF4LnRvb2xzLmNvbGFiX3RwdSUwQSUwQWpheC50b29scy5jb2xhYl90cHUuc2V0dXBfdHB1KCklMEFpbXBvcnQlMjBqYXg=",highlighted:`<span class="hljs-keyword">import</span> jax.tools.colab_tpu
jax.tools.colab_tpu.setup_tpu()
<span class="hljs-keyword">import</span> jax`,wrap:!1}}),_=new b({props:{code:"bnVtX2RldmljZXMlMjAlM0QlMjBqYXguZGV2aWNlX2NvdW50KCklMEFkZXZpY2VfdHlwZSUyMCUzRCUyMGpheC5kZXZpY2VzKCklNUIwJTVELmRldmljZV9raW5kJTBBJTBBcHJpbnQoZiUyMkZvdW5kJTIwJTdCbnVtX2RldmljZXMlN0QlMjBKQVglMjBkZXZpY2VzJTIwb2YlMjB0eXBlJTIwJTdCZGV2aWNlX3R5cGUlN0QuJTIyKSUwQWFzc2VydCUyMCglMEElMjAlMjAlMjAlMjAlMjJUUFUlMjIlMjBpbiUyMGRldmljZV90eXBlJTBBKSUyQyUyMCUyMkF2YWlsYWJsZSUyMGRldmljZSUyMGlzJTIwbm90JTIwYSUyMFRQVSUyQyUyMHBsZWFzZSUyMHNlbGVjdCUyMFRQVSUyMGZyb20lMjBFZGl0JTIwJTNFJTIwTm90ZWJvb2slMjBzZXR0aW5ncyUyMCUzRSUyMEhhcmR3YXJlJTIwYWNjZWxlcmF0b3IlMjI=",highlighted:`num_devices = jax.device_count()
device_type = jax.devices()[<span class="hljs-number">0</span>].device_kind
<span class="hljs-built_in">print</span>(<span class="hljs-string">f&quot;Found <span class="hljs-subst">{num_devices}</span> JAX devices of type <span class="hljs-subst">{device_type}</span>.&quot;</span>)
<span class="hljs-keyword">assert</span> (
<span class="hljs-string">&quot;TPU&quot;</span> <span class="hljs-keyword">in</span> device_type
), <span class="hljs-string">&quot;Available device is not a TPU, please select TPU from Edit &gt; Notebook settings &gt; Hardware accelerator&quot;</span>`,wrap:!1}}),I=new b({props:{code:"Rm91bmQlMjA4JTIwSkFYJTIwZGV2aWNlcyUyMG9mJTIwdHlwZSUyMENsb3VkJTIwVFBVLg==",highlighted:'Found <span class="hljs-number">8</span> JAX devices of <span class="hljs-built_in">type</span> Cloud TPU.',wrap:!1}}),x=new b({props:{code:"aW1wb3J0JTIwbnVtcHklMjBhcyUyMG5wJTBBaW1wb3J0JTIwamF4JTBBaW1wb3J0JTIwamF4Lm51bXB5JTIwYXMlMjBqbnAlMEElMEFmcm9tJTIwcGF0aGxpYiUyMGltcG9ydCUyMFBhdGglMEFmcm9tJTIwamF4JTIwaW1wb3J0JTIwcG1hcCUwQWZyb20lMjBmbGF4LmpheF91dGlscyUyMGltcG9ydCUyMHJlcGxpY2F0ZSUwQWZyb20lMjBmbGF4LnRyYWluaW5nLmNvbW1vbl91dGlscyUyMGltcG9ydCUyMHNoYXJkJTBBZnJvbSUyMFBJTCUyMGltcG9ydCUyMEltYWdlJTBBJTBBZnJvbSUyMGh1Z2dpbmdmYWNlX2h1YiUyMGltcG9ydCUyMG5vdGVib29rX2xvZ2luJTBBZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMEZsYXhTdGFibGVEaWZmdXNpb25QaXBlbGluZQ==",highlighted:`<span class="hljs-keyword">import</span> numpy <span class="hljs-keyword">as</span> np
<span class="hljs-keyword">import</span> jax
<span class="hljs-keyword">import</span> jax.numpy <span class="hljs-keyword">as</span> jnp
<span class="hljs-keyword">from</span> pathlib <span class="hljs-keyword">import</span> Path
<span class="hljs-keyword">from</span> jax <span class="hljs-keyword">import</span> pmap
<span class="hljs-keyword">from</span> flax.jax_utils <span class="hljs-keyword">import</span> replicate
<span class="hljs-keyword">from</span> flax.training.common_utils <span class="hljs-keyword">import</span> shard
<span class="hljs-keyword">from</span> PIL <span class="hljs-keyword">import</span> Image
<span class="hljs-keyword">from</span> huggingface_hub <span class="hljs-keyword">import</span> notebook_login
<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> FlaxStableDiffusionPipeline`,wrap:!1}}),v=new J({props:{title:"λͺ¨λΈ 뢈러였기",local:"λͺ¨λΈ-뢈러였기",headingTag:"h2"}}),Z=new b({props:{code:"ZHR5cGUlMjAlM0QlMjBqbnAuYmZsb2F0MTY=",highlighted:"dtype = jnp.bfloat16",wrap:!1}}),X=new b({props:{code:"cGlwZWxpbmUlMkMlMjBwYXJhbXMlMjAlM0QlMjBGbGF4U3RhYmxlRGlmZnVzaW9uUGlwZWxpbmUuZnJvbV9wcmV0cmFpbmVkKCUwQSUyMCUyMCUyMCUyMCUyMkNvbXBWaXMlMkZzdGFibGUtZGlmZnVzaW9uLXYxLTQlMjIlMkMlMEElMjAlMjAlMjAlMjB2YXJpYW50JTNEJTIyYmYxNiUyMiUyQyUwQSUyMCUyMCUyMCUyMGR0eXBlJTNEZHR5cGUlMkMlMEEp",highlighted:`pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
<span class="hljs-string">&quot;CompVis/stable-diffusion-v1-4&quot;</span>,
variant=<span class="hljs-string">&quot;bf16&quot;</span>,
dtype=dtype,
)`,wrap:!1}}),H=new J({props:{title:"μΆ”λ‘ ",local:"μΆ”λ‘ ",headingTag:"h2"}}),Q=new b({props:{code:"cHJvbXB0JTIwJTNEJTIwJTIyQSUyMGNpbmVtYXRpYyUyMGZpbG0lMjBzdGlsbCUyMG9mJTIwTW9yZ2FuJTIwRnJlZW1hbiUyMHN0YXJyaW5nJTIwYXMlMjBKaW1pJTIwSGVuZHJpeCUyQyUyMHBvcnRyYWl0JTJDJTIwNDBtbSUyMGxlbnMlMkMlMjBzaGFsbG93JTIwZGVwdGglMjBvZiUyMGZpZWxkJTJDJTIwY2xvc2UlMjB1cCUyQyUyMHNwbGl0JTIwbGlnaHRpbmclMkMlMjBjaW5lbWF0aWMlMjIlMEFwcm9tcHQlMjAlM0QlMjAlNUJwcm9tcHQlNUQlMjAqJTIwamF4LmRldmljZV9jb3VudCgpJTBBcHJvbXB0X2lkcyUyMCUzRCUyMHBpcGVsaW5lLnByZXBhcmVfaW5wdXRzKHByb21wdCklMEFwcm9tcHRfaWRzLnNoYXBl",highlighted:`prompt = <span class="hljs-string">&quot;A cinematic film still of Morgan Freeman starring as Jimi Hendrix, portrait, 40mm lens, shallow depth of field, close up, split lighting, cinematic&quot;</span>
prompt = [prompt] * jax.device_count()
prompt_ids = pipeline.prepare_inputs(prompt)
prompt_ids.shape`,wrap:!1}}),V=new b({props:{code:"KDglMkMlMjA3Nyk=",highlighted:'(<span class="hljs-number">8</span>, <span class="hljs-number">77</span>)',wrap:!1}}),z=new J({props:{title:"볡사(Replication) 및 μ •λ ¬ν™”",local:"볡사replication-및-μ •λ ¬ν™”",headingTag:"h3"}}),N=new b({props:{code:"cF9wYXJhbXMlMjAlM0QlMjByZXBsaWNhdGUocGFyYW1zKQ==",highlighted:"p_params = replicate(params)",wrap:!1}}),S=new b({props:{code:"cHJvbXB0X2lkcyUyMCUzRCUyMHNoYXJkKHByb21wdF9pZHMpJTBBcHJvbXB0X2lkcy5zaGFwZQ==",highlighted:`prompt_ids = shard(prompt_ids)
prompt_ids.shape`,wrap:!1}}),L=new b({props:{code:"KDglMkMlMjAxJTJDJTIwNzcp",highlighted:'(<span class="hljs-number">8</span>, <span class="hljs-number">1</span>, <span class="hljs-number">77</span>)',wrap:!1}}),q=new b({props:{code:"ZGVmJTIwY3JlYXRlX2tleShzZWVkJTNEMCklM0ElMEElMjAlMjAlMjAlMjByZXR1cm4lMjBqYXgucmFuZG9tLlBSTkdLZXkoc2VlZCk=",highlighted:`<span class="hljs-keyword">def</span> <span class="hljs-title function_">create_key</span>(<span class="hljs-params">seed=<span class="hljs-number">0</span></span>):
<span class="hljs-keyword">return</span> jax.random.PRNGKey(seed)`,wrap:!1}}),K=new b({props:{code:"cm5nJTIwJTNEJTIwY3JlYXRlX2tleSgwKSUwQXJuZyUyMCUzRCUyMGpheC5yYW5kb20uc3BsaXQocm5nJTJDJTIwamF4LmRldmljZV9jb3VudCgpKQ==",highlighted:`rng = create_key(<span class="hljs-number">0</span>)
rng = jax.random.split(rng, jax.device_count())`,wrap:!1}}),le=new b({props:{code:"JTI1JTI1dGltZSUwQWltYWdlcyUyMCUzRCUyMHBpcGVsaW5lKHByb21wdF9pZHMlMkMlMjBwX3BhcmFtcyUyQyUyMHJuZyUyQyUyMGppdCUzRFRydWUpJTVCMCU1RA==",highlighted:`%%time
images = pipeline(prompt_ids, p_params, rng, <span class="hljs-attribute">jit</span>=<span class="hljs-literal">True</span>)[0]`,wrap:!1}}),se=new b({props:{code:"Q1BVJTIwdGltZXMlM0ElMjB1c2VyJTIwNTYuMiUyMHMlMkMlMjBzeXMlM0ElMjA0Mi41JTIwcyUyQyUyMHRvdGFsJTNBJTIwMW1pbiUyMDM4cyUwQVdhbGwlMjB0aW1lJTNBJTIwMW1pbiUyMDI5cw==",highlighted:`CPU times: user <span class="hljs-number">56.2</span> s, sys: <span class="hljs-number">42.5</span> s, total: 1<span class="hljs-built_in">min</span> 38s
Wall time: 1<span class="hljs-built_in">min</span> 29s`,wrap:!1}}),pe=new b({props:{code:"aW1hZ2VzJTIwJTNEJTIwaW1hZ2VzLnJlc2hhcGUoKGltYWdlcy5zaGFwZSU1QjAlNUQlMjAqJTIwaW1hZ2VzLnNoYXBlJTVCMSU1RCUyQyklMjAlMkIlMjBpbWFnZXMuc2hhcGUlNUItMyUzQSU1RCklMEFpbWFnZXMlMjAlM0QlMjBwaXBlbGluZS5udW1weV90b19waWwoaW1hZ2VzKQ==",highlighted:`images = images.reshape((images.shape[<span class="hljs-number">0</span>] * images.shape[<span class="hljs-number">1</span>],) + images.shape[-<span class="hljs-number">3</span>:])
images = pipeline.numpy_to_pil(images)`,wrap:!1}}),ne=new J({props:{title:"μ‹œκ°ν™”",local:"μ‹œκ°ν™”",headingTag:"h3"}}),me=new b({props:{code:"ZGVmJTIwaW1hZ2VfZ3JpZChpbWdzJTJDJTIwcm93cyUyQyUyMGNvbHMpJTNBJTBBJTIwJTIwJTIwJTIwdyUyQyUyMGglMjAlM0QlMjBpbWdzJTVCMCU1RC5zaXplJTBBJTIwJTIwJTIwJTIwZ3JpZCUyMCUzRCUyMEltYWdlLm5ldyglMjJSR0IlMjIlMkMlMjBzaXplJTNEKGNvbHMlMjAqJTIwdyUyQyUyMHJvd3MlMjAqJTIwaCkpJTBBJTIwJTIwJTIwJTIwZm9yJTIwaSUyQyUyMGltZyUyMGluJTIwZW51bWVyYXRlKGltZ3MpJTNBJTBBJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwZ3JpZC5wYXN0ZShpbWclMkMlMjBib3glM0QoaSUyMCUyNSUyMGNvbHMlMjAqJTIwdyUyQyUyMGklMjAlMkYlMkYlMjBjb2xzJTIwKiUyMGgpKSUwQSUyMCUyMCUyMCUyMHJldHVybiUyMGdyaWQ=",highlighted:`<span class="hljs-keyword">def</span> <span class="hljs-title function_">image_grid</span>(<span class="hljs-params">imgs, rows, cols</span>):
w, h = imgs[<span class="hljs-number">0</span>].size
grid = Image.new(<span class="hljs-string">&quot;RGB&quot;</span>, size=(cols * w, rows * h))
<span class="hljs-keyword">for</span> i, img <span class="hljs-keyword">in</span> <span class="hljs-built_in">enumerate</span>(imgs):
grid.paste(img, box=(i % cols * w, i // cols * h))
<span class="hljs-keyword">return</span> grid`,wrap:!1}}),Me=new b({props:{code:"aW1hZ2VfZ3JpZChpbWFnZXMlMkMlMjAyJTJDJTIwNCk=",highlighted:'image_grid(images, <span class="hljs-number">2</span>, <span class="hljs-number">4</span>)',wrap:!1}}),oe=new J({props:{title:"λ‹€λ₯Έ ν”„λ‘¬ν”„νŠΈ μ‚¬μš©",local:"λ‹€λ₯Έ-ν”„λ‘¬ν”„νŠΈ-μ‚¬μš©",headingTag:"h2"}}),re=new b({props:{code:"cHJvbXB0cyUyMCUzRCUyMCU1QiUwQSUyMCUyMCUyMCUyMCUyMkxhYnJhZG9yJTIwaW4lMjB0aGUlMjBzdHlsZSUyMG9mJTIwSG9rdXNhaSUyMiUyQyUwQSUyMCUyMCUyMCUyMCUyMlBhaW50aW5nJTIwb2YlMjBhJTIwc3F1aXJyZWwlMjBza2F0aW5nJTIwaW4lMjBOZXclMjBZb3JrJTIyJTJDJTBBJTIwJTIwJTIwJTIwJTIySEFMLTkwMDAlMjBpbiUyMHRoZSUyMHN0eWxlJTIwb2YlMjBWYW4lMjBHb2doJTIyJTJDJTBBJTIwJTIwJTIwJTIwJTIyVGltZXMlMjBTcXVhcmUlMjB1bmRlciUyMHdhdGVyJTJDJTIwd2l0aCUyMGZpc2glMjBhbmQlMjBhJTIwZG9scGhpbiUyMHN3aW1taW5nJTIwYXJvdW5kJTIyJTJDJTBBJTIwJTIwJTIwJTIwJTIyQW5jaWVudCUyMFJvbWFuJTIwZnJlc2NvJTIwc2hvd2luZyUyMGElMjBtYW4lMjB3b3JraW5nJTIwb24lMjBoaXMlMjBsYXB0b3AlMjIlMkMlMEElMjAlMjAlMjAlMjAlMjJDbG9zZS11cCUyMHBob3RvZ3JhcGglMjBvZiUyMHlvdW5nJTIwYmxhY2slMjB3b21hbiUyMGFnYWluc3QlMjB1cmJhbiUyMGJhY2tncm91bmQlMkMlMjBoaWdoJTIwcXVhbGl0eSUyQyUyMGJva2VoJTIyJTJDJTBBJTIwJTIwJTIwJTIwJTIyQXJtY2hhaXIlMjBpbiUyMHRoZSUyMHNoYXBlJTIwb2YlMjBhbiUyMGF2b2NhZG8lMjIlMkMlMEElMjAlMjAlMjAlMjAlMjJDbG93biUyMGFzdHJvbmF1dCUyMGluJTIwc3BhY2UlMkMlMjB3aXRoJTIwRWFydGglMjBpbiUyMHRoZSUyMGJhY2tncm91bmQlMjIlMkMlMEElNUQ=",highlighted:`prompts = [
<span class="hljs-string">&quot;Labrador in the style of Hokusai&quot;</span>,
<span class="hljs-string">&quot;Painting of a squirrel skating in New York&quot;</span>,
<span class="hljs-string">&quot;HAL-9000 in the style of Van Gogh&quot;</span>,
<span class="hljs-string">&quot;Times Square under water, with fish and a dolphin swimming around&quot;</span>,
<span class="hljs-string">&quot;Ancient Roman fresco showing a man working on his laptop&quot;</span>,
<span class="hljs-string">&quot;Close-up photograph of young black woman against urban background, high quality, bokeh&quot;</span>,
<span class="hljs-string">&quot;Armchair in the shape of an avocado&quot;</span>,
<span class="hljs-string">&quot;Clown astronaut in space, with Earth in the background&quot;</span>,
]`,wrap:!1}}),be=new b({props:{code:"cHJvbXB0X2lkcyUyMCUzRCUyMHBpcGVsaW5lLnByZXBhcmVfaW5wdXRzKHByb21wdHMpJTBBcHJvbXB0X2lkcyUyMCUzRCUyMHNoYXJkKHByb21wdF9pZHMpJTBBJTBBaW1hZ2VzJTIwJTNEJTIwcGlwZWxpbmUocHJvbXB0X2lkcyUyQyUyMHBfcGFyYW1zJTJDJTIwcm5nJTJDJTIwaml0JTNEVHJ1ZSkuaW1hZ2VzJTBBaW1hZ2VzJTIwJTNEJTIwaW1hZ2VzLnJlc2hhcGUoKGltYWdlcy5zaGFwZSU1QjAlNUQlMjAqJTIwaW1hZ2VzLnNoYXBlJTVCMSU1RCUyQyklMjAlMkIlMjBpbWFnZXMuc2hhcGUlNUItMyUzQSU1RCklMEFpbWFnZXMlMjAlM0QlMjBwaXBlbGluZS5udW1weV90b19waWwoaW1hZ2VzKSUwQSUwQWltYWdlX2dyaWQoaW1hZ2VzJTJDJTIwMiUyQyUyMDQp",highlighted:`prompt_ids = pipeline.prepare_inputs(prompts)
prompt_ids = shard(prompt_ids)
images = pipeline(prompt_ids, p_params, rng, jit=<span class="hljs-literal">True</span>).images
images = images.reshape((images.shape[<span class="hljs-number">0</span>] * images.shape[<span class="hljs-number">1</span>],) + images.shape[-<span class="hljs-number">3</span>:])
images = pipeline.numpy_to_pil(images)
image_grid(images, <span class="hljs-number">2</span>, <span class="hljs-number">4</span>)`,wrap:!1}}),Je=new J({props:{title:"병렬화(parallelization)λŠ” μ–΄λ–»κ²Œ μž‘λ™ν•˜λŠ”κ°€?",local:"병렬화parallelizationλŠ”-μ–΄λ–»κ²Œ-μž‘λ™ν•˜λŠ”κ°€",headingTag:"h2"}}),he=new b({props:{code:"cF9nZW5lcmF0ZSUyMCUzRCUyMHBtYXAocGlwZWxpbmUuX2dlbmVyYXRlKQ==",highlighted:"p_generate = pmap(pipeline._generate)",wrap:!1}}),_e=new b({props:{code:"JTI1JTI1dGltZSUwQWltYWdlcyUyMCUzRCUyMHBfZ2VuZXJhdGUocHJvbXB0X2lkcyUyQyUyMHBfcGFyYW1zJTJDJTIwcm5nKSUwQWltYWdlcyUyMCUzRCUyMGltYWdlcy5ibG9ja191bnRpbF9yZWFkeSgpJTBBaW1hZ2VzLnNoYXBl",highlighted:`%%time
images = p<span class="hljs-constructor">_generate(<span class="hljs-params">prompt_ids</span>, <span class="hljs-params">p_params</span>, <span class="hljs-params">rng</span>)</span>
images = images.block<span class="hljs-constructor">_until_ready()</span>
images.shape`,wrap:!1}}),Ie=new b({props:{code:"Q1BVJTIwdGltZXMlM0ElMjB1c2VyJTIwMW1pbiUyMDE1cyUyQyUyMHN5cyUzQSUyMDE4LjIlMjBzJTJDJTIwdG90YWwlM0ElMjAxbWluJTIwMzRzJTBBV2FsbCUyMHRpbWUlM0ElMjAxbWluJTIwMTVz",highlighted:`CPU times: user 1<span class="hljs-built_in">min</span> 15s, sys: <span class="hljs-number">18.2</span> s, total: 1<span class="hljs-built_in">min</span> 34s
Wall time: 1<span class="hljs-built_in">min</span> 15s`,wrap:!1}}),ke=new b({props:{code:"aW1hZ2VzLnNoYXBl",highlighted:"images.shape",wrap:!1}}),xe=new b({props:{code:"KDglMkMlMjAxJTJDJTIwNTEyJTJDJTIwNTEyJTJDJTIwMyk=",highlighted:'(<span class="hljs-number">8</span>, <span class="hljs-number">1</span>, <span class="hljs-number">512</span>, <span class="hljs-number">512</span>, <span class="hljs-number">3</span>)',wrap:!1}}),Ge=new Ll({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/ko/using-diffusers/stable_diffusion_jax_how_to.md"}}),{c(){u=n("meta"),Xe=a(),Ze=n("p"),He=a(),m(j.$$.fragment),Fe=a(),m($.$$.fragment),Ye=a(),c=n("p"),c.innerHTML=ll,Qe=a(),T=n("p"),T.innerHTML=sl,Ve=a(),d=n("p"),d.textContent=al,ze=a(),h=n("p"),h.textContent=pl,Re=a(),m(U.$$.fragment),Ne=a(),g=n("p"),g.textContent=nl,Se=a(),m(B.$$.fragment),Le=a(),m(C.$$.fragment),Ae=a(),m(_.$$.fragment),Pe=a(),m(I.$$.fragment),Ee=a(),k=n("p"),k.textContent=il,qe=a(),m(x.$$.fragment),De=a(),m(v.$$.fragment),Ke=a(),G=n("p"),G.textContent=ml,Oe=a(),m(Z.$$.fragment),et=a(),W=n("p"),W.textContent=Ml,tt=a(),m(X.$$.fragment),lt=a(),m(H.$$.fragment),st=a(),F=n("p"),F.textContent=fl,at=a(),Y=n("p"),Y.innerHTML=ol,pt=a(),m(Q.$$.fragment),nt=a(),m(V.$$.fragment),it=a(),m(z.$$.fragment),mt=a(),R=n("p"),R.innerHTML=yl,Mt=a(),m(N.$$.fragment),ft=a(),m(S.$$.fragment),ot=a(),m(L.$$.fragment),yt=a(),A=n("p"),A.innerHTML=wl,wt=a(),P=n("p"),P.textContent=rl,rt=a(),E=n("p"),E.textContent=bl,bt=a(),m(q.$$.fragment),ut=a(),D=n("p"),D.textContent=ul,Jt=a(),m(K.$$.fragment),jt=a(),O=n("p"),O.textContent=Jl,$t=a(),ee=n("p"),ee.innerHTML=jl,ct=a(),te=n("p"),te.textContent=$l,Tt=a(),m(le.$$.fragment),dt=a(),m(se.$$.fragment),ht=a(),ae=n("p"),ae.innerHTML=cl,Ut=a(),m(pe.$$.fragment),gt=a(),m(ne.$$.fragment),Bt=a(),ie=n("p"),ie.textContent=Tl,Ct=a(),m(me.$$.fragment),_t=a(),m(Me.$$.fragment),It=a(),fe=n("p"),fe.innerHTML=dl,kt=a(),m(oe.$$.fragment),xt=a(),ye=n("p"),ye.textContent=hl,vt=a(),we=n("p"),we.textContent=Ul,Gt=a(),m(re.$$.fragment),Zt=a(),m(be.$$.fragment),Wt=a(),ue=n("p"),ue.innerHTML=gl,Xt=a(),m(Je.$$.fragment),Ht=a(),je=n("p"),je.innerHTML=Bl,Ft=a(),$e=n("p"),$e.innerHTML=Cl,Yt=a(),ce=n("p"),ce.innerHTML=_l,Qt=a(),Te=n("ul"),Te.innerHTML=Il,Vt=a(),de=n("p"),de.innerHTML=kl,zt=a(),m(he.$$.fragment),Rt=a(),Ue=n("p"),Ue.innerHTML=xl,Nt=a(),ge=n("ul"),ge.innerHTML=vl,St=a(),Be=n("p"),Be.innerHTML=Gl,Lt=a(),Ce=n("p"),Ce.textContent=Zl,At=a(),m(_e.$$.fragment),Pt=a(),m(Ie.$$.fragment),Et=a(),m(ke.$$.fragment),qt=a(),m(xe.$$.fragment),Dt=a(),ve=n("p"),ve.innerHTML=Wl,Kt=a(),m(Ge.$$.fragment),Ot=a(),We=n("p"),this.h()},l(e){const t=Rl("svelte-u9bgzb",document.head);u=i(t,"META",{name:!0,content:!0}),t.forEach(l),Xe=p(e),Ze=i(e,"P",{}),Xl(Ze).forEach(l),He=p(e),M(j.$$.fragment,e),Fe=p(e),M($.$$.fragment,e),Ye=p(e),c=i(e,"P",{"data-svelte-h":!0}),r(c)!=="svelte-1gxtaje"&&(c.innerHTML=ll),Qe=p(e),T=i(e,"P",{"data-svelte-h":!0}),r(T)!=="svelte-1bumom2"&&(T.innerHTML=sl),Ve=p(e),d=i(e,"P",{"data-svelte-h":!0}),r(d)!=="svelte-17mi6da"&&(d.textContent=al),ze=p(e),h=i(e,"P",{"data-svelte-h":!0}),r(h)!=="svelte-hjgegg"&&(h.textContent=pl),Re=p(e),M(U.$$.fragment,e),Ne=p(e),g=i(e,"P",{"data-svelte-h":!0}),r(g)!=="svelte-17ncbx5"&&(g.textContent=nl),Se=p(e),M(B.$$.fragment,e),Le=p(e),M(C.$$.fragment,e),Ae=p(e),M(_.$$.fragment,e),Pe=p(e),M(I.$$.fragment,e),Ee=p(e),k=i(e,"P",{"data-svelte-h":!0}),r(k)!=="svelte-95k7v6"&&(k.textContent=il),qe=p(e),M(x.$$.fragment,e),De=p(e),M(v.$$.fragment,e),Ke=p(e),G=i(e,"P",{"data-svelte-h":!0}),r(G)!=="svelte-71fx65"&&(G.textContent=ml),Oe=p(e),M(Z.$$.fragment,e),et=p(e),W=i(e,"P",{"data-svelte-h":!0}),r(W)!=="svelte-1uvv4fp"&&(W.textContent=Ml),tt=p(e),M(X.$$.fragment,e),lt=p(e),M(H.$$.fragment,e),st=p(e),F=i(e,"P",{"data-svelte-h":!0}),r(F)!=="svelte-7wabl0"&&(F.textContent=fl),at=p(e),Y=i(e,"P",{"data-svelte-h":!0}),r(Y)!=="svelte-13u2ylm"&&(Y.innerHTML=ol),pt=p(e),M(Q.$$.fragment,e),nt=p(e),M(V.$$.fragment,e),it=p(e),M(z.$$.fragment,e),mt=p(e),R=i(e,"P",{"data-svelte-h":!0}),r(R)!=="svelte-qzdgm1"&&(R.innerHTML=yl),Mt=p(e),M(N.$$.fragment,e),ft=p(e),M(S.$$.fragment,e),ot=p(e),M(L.$$.fragment,e),yt=p(e),A=i(e,"P",{"data-svelte-h":!0}),r(A)!=="svelte-i4vbvw"&&(A.innerHTML=wl),wt=p(e),P=i(e,"P",{"data-svelte-h":!0}),r(P)!=="svelte-1ghn5le"&&(P.textContent=rl),rt=p(e),E=i(e,"P",{"data-svelte-h":!0}),r(E)!=="svelte-1lcan7m"&&(E.textContent=bl),bt=p(e),M(q.$$.fragment,e),ut=p(e),D=i(e,"P",{"data-svelte-h":!0}),r(D)!=="svelte-1ngdgb0"&&(D.textContent=ul),Jt=p(e),M(K.$$.fragment,e),jt=p(e),O=i(e,"P",{"data-svelte-h":!0}),r(O)!=="svelte-103e30a"&&(O.textContent=Jl),$t=p(e),ee=i(e,"P",{"data-svelte-h":!0}),r(ee)!=="svelte-1sbamam"&&(ee.innerHTML=jl),ct=p(e),te=i(e,"P",{"data-svelte-h":!0}),r(te)!=="svelte-15bwfw"&&(te.textContent=$l),Tt=p(e),M(le.$$.fragment,e),dt=p(e),M(se.$$.fragment,e),ht=p(e),ae=i(e,"P",{"data-svelte-h":!0}),r(ae)!=="svelte-1egnab9"&&(ae.innerHTML=cl),Ut=p(e),M(pe.$$.fragment,e),gt=p(e),M(ne.$$.fragment,e),Bt=p(e),ie=i(e,"P",{"data-svelte-h":!0}),r(ie)!=="svelte-1e76x5r"&&(ie.textContent=Tl),Ct=p(e),M(me.$$.fragment,e),_t=p(e),M(Me.$$.fragment,e),It=p(e),fe=i(e,"P",{"data-svelte-h":!0}),r(fe)!=="svelte-e1s2k"&&(fe.innerHTML=dl),kt=p(e),M(oe.$$.fragment,e),xt=p(e),ye=i(e,"P",{"data-svelte-h":!0}),r(ye)!=="svelte-1buolzv"&&(ye.textContent=hl),vt=p(e),we=i(e,"P",{"data-svelte-h":!0}),r(we)!=="svelte-1fscvnd"&&(we.textContent=Ul),Gt=p(e),M(re.$$.fragment,e),Zt=p(e),M(be.$$.fragment,e),Wt=p(e),ue=i(e,"P",{"data-svelte-h":!0}),r(ue)!=="svelte-1nea4f2"&&(ue.innerHTML=gl),Xt=p(e),M(Je.$$.fragment,e),Ht=p(e),je=i(e,"P",{"data-svelte-h":!0}),r(je)!=="svelte-1780g9f"&&(je.innerHTML=Bl),Ft=p(e),$e=i(e,"P",{"data-svelte-h":!0}),r($e)!=="svelte-d62kvx"&&($e.innerHTML=Cl),Yt=p(e),ce=i(e,"P",{"data-svelte-h":!0}),r(ce)!=="svelte-1ykkmgr"&&(ce.innerHTML=_l),Qt=p(e),Te=i(e,"UL",{"data-svelte-h":!0}),r(Te)!=="svelte-2aqcsn"&&(Te.innerHTML=Il),Vt=p(e),de=i(e,"P",{"data-svelte-h":!0}),r(de)!=="svelte-mu1la1"&&(de.innerHTML=kl),zt=p(e),M(he.$$.fragment,e),Rt=p(e),Ue=i(e,"P",{"data-svelte-h":!0}),r(Ue)!=="svelte-52cpde"&&(Ue.innerHTML=xl),Nt=p(e),ge=i(e,"UL",{"data-svelte-h":!0}),r(ge)!=="svelte-11847k7"&&(ge.innerHTML=vl),St=p(e),Be=i(e,"P",{"data-svelte-h":!0}),r(Be)!=="svelte-1x822d1"&&(Be.innerHTML=Gl),Lt=p(e),Ce=i(e,"P",{"data-svelte-h":!0}),r(Ce)!=="svelte-1ukvcj9"&&(Ce.textContent=Zl),At=p(e),M(_e.$$.fragment,e),Pt=p(e),M(Ie.$$.fragment,e),Et=p(e),M(ke.$$.fragment,e),qt=p(e),M(xe.$$.fragment,e),Dt=p(e),ve=i(e,"P",{"data-svelte-h":!0}),r(ve)!=="svelte-1s6my01"&&(ve.innerHTML=Wl),Kt=p(e),M(Ge.$$.fragment,e),Ot=p(e),We=i(e,"P",{}),Xl(We).forEach(l),this.h()},h(){Hl(u,"name","hf:doc:metadata"),Hl(u,"content",Pl)},m(e,t){Nl(document.head,u),s(e,Xe,t),s(e,Ze,t),s(e,He,t),f(j,e,t),s(e,Fe,t),f($,e,t),s(e,Ye,t),s(e,c,t),s(e,Qe,t),s(e,T,t),s(e,Ve,t),s(e,d,t),s(e,ze,t),s(e,h,t),s(e,Re,t),f(U,e,t),s(e,Ne,t),s(e,g,t),s(e,Se,t),f(B,e,t),s(e,Le,t),f(C,e,t),s(e,Ae,t),f(_,e,t),s(e,Pe,t),f(I,e,t),s(e,Ee,t),s(e,k,t),s(e,qe,t),f(x,e,t),s(e,De,t),f(v,e,t),s(e,Ke,t),s(e,G,t),s(e,Oe,t),f(Z,e,t),s(e,et,t),s(e,W,t),s(e,tt,t),f(X,e,t),s(e,lt,t),f(H,e,t),s(e,st,t),s(e,F,t),s(e,at,t),s(e,Y,t),s(e,pt,t),f(Q,e,t),s(e,nt,t),f(V,e,t),s(e,it,t),f(z,e,t),s(e,mt,t),s(e,R,t),s(e,Mt,t),f(N,e,t),s(e,ft,t),f(S,e,t),s(e,ot,t),f(L,e,t),s(e,yt,t),s(e,A,t),s(e,wt,t),s(e,P,t),s(e,rt,t),s(e,E,t),s(e,bt,t),f(q,e,t),s(e,ut,t),s(e,D,t),s(e,Jt,t),f(K,e,t),s(e,jt,t),s(e,O,t),s(e,$t,t),s(e,ee,t),s(e,ct,t),s(e,te,t),s(e,Tt,t),f(le,e,t),s(e,dt,t),f(se,e,t),s(e,ht,t),s(e,ae,t),s(e,Ut,t),f(pe,e,t),s(e,gt,t),f(ne,e,t),s(e,Bt,t),s(e,ie,t),s(e,Ct,t),f(me,e,t),s(e,_t,t),f(Me,e,t),s(e,It,t),s(e,fe,t),s(e,kt,t),f(oe,e,t),s(e,xt,t),s(e,ye,t),s(e,vt,t),s(e,we,t),s(e,Gt,t),f(re,e,t),s(e,Zt,t),f(be,e,t),s(e,Wt,t),s(e,ue,t),s(e,Xt,t),f(Je,e,t),s(e,Ht,t),s(e,je,t),s(e,Ft,t),s(e,$e,t),s(e,Yt,t),s(e,ce,t),s(e,Qt,t),s(e,Te,t),s(e,Vt,t),s(e,de,t),s(e,zt,t),f(he,e,t),s(e,Rt,t),s(e,Ue,t),s(e,Nt,t),s(e,ge,t),s(e,St,t),s(e,Be,t),s(e,Lt,t),s(e,Ce,t),s(e,At,t),f(_e,e,t),s(e,Pt,t),f(Ie,e,t),s(e,Et,t),f(ke,e,t),s(e,qt,t),f(xe,e,t),s(e,Dt,t),s(e,ve,t),s(e,Kt,t),f(Ge,e,t),s(e,Ot,t),s(e,We,t),el=!0},p:Yl,i(e){el||(o(j.$$.fragment,e),o($.$$.fragment,e),o(U.$$.fragment,e),o(B.$$.fragment,e),o(C.$$.fragment,e),o(_.$$.fragment,e),o(I.$$.fragment,e),o(x.$$.fragment,e),o(v.$$.fragment,e),o(Z.$$.fragment,e),o(X.$$.fragment,e),o(H.$$.fragment,e),o(Q.$$.fragment,e),o(V.$$.fragment,e),o(z.$$.fragment,e),o(N.$$.fragment,e),o(S.$$.fragment,e),o(L.$$.fragment,e),o(q.$$.fragment,e),o(K.$$.fragment,e),o(le.$$.fragment,e),o(se.$$.fragment,e),o(pe.$$.fragment,e),o(ne.$$.fragment,e),o(me.$$.fragment,e),o(Me.$$.fragment,e),o(oe.$$.fragment,e),o(re.$$.fragment,e),o(be.$$.fragment,e),o(Je.$$.fragment,e),o(he.$$.fragment,e),o(_e.$$.fragment,e),o(Ie.$$.fragment,e),o(ke.$$.fragment,e),o(xe.$$.fragment,e),o(Ge.$$.fragment,e),el=!0)},o(e){y(j.$$.fragment,e),y($.$$.fragment,e),y(U.$$.fragment,e),y(B.$$.fragment,e),y(C.$$.fragment,e),y(_.$$.fragment,e),y(I.$$.fragment,e),y(x.$$.fragment,e),y(v.$$.fragment,e),y(Z.$$.fragment,e),y(X.$$.fragment,e),y(H.$$.fragment,e),y(Q.$$.fragment,e),y(V.$$.fragment,e),y(z.$$.fragment,e),y(N.$$.fragment,e),y(S.$$.fragment,e),y(L.$$.fragment,e),y(q.$$.fragment,e),y(K.$$.fragment,e),y(le.$$.fragment,e),y(se.$$.fragment,e),y(pe.$$.fragment,e),y(ne.$$.fragment,e),y(me.$$.fragment,e),y(Me.$$.fragment,e),y(oe.$$.fragment,e),y(re.$$.fragment,e),y(be.$$.fragment,e),y(Je.$$.fragment,e),y(he.$$.fragment,e),y(_e.$$.fragment,e),y(Ie.$$.fragment,e),y(ke.$$.fragment,e),y(xe.$$.fragment,e),y(Ge.$$.fragment,e),el=!1},d(e){e&&(l(Xe),l(Ze),l(He),l(Fe),l(Ye),l(c),l(Qe),l(T),l(Ve),l(d),l(ze),l(h),l(Re),l(Ne),l(g),l(Se),l(Le),l(Ae),l(Pe),l(Ee),l(k),l(qe),l(De),l(Ke),l(G),l(Oe),l(et),l(W),l(tt),l(lt),l(st),l(F),l(at),l(Y),l(pt),l(nt),l(it),l(mt),l(R),l(Mt),l(ft),l(ot),l(yt),l(A),l(wt),l(P),l(rt),l(E),l(bt),l(ut),l(D),l(Jt),l(jt),l(O),l($t),l(ee),l(ct),l(te),l(Tt),l(dt),l(ht),l(ae),l(Ut),l(gt),l(Bt),l(ie),l(Ct),l(_t),l(It),l(fe),l(kt),l(xt),l(ye),l(vt),l(we),l(Gt),l(Zt),l(Wt),l(ue),l(Xt),l(Ht),l(je),l(Ft),l($e),l(Yt),l(ce),l(Qt),l(Te),l(Vt),l(de),l(zt),l(Rt),l(Ue),l(Nt),l(ge),l(St),l(Be),l(Lt),l(Ce),l(At),l(Pt),l(Et),l(qt),l(Dt),l(ve),l(Kt),l(Ot),l(We)),l(u),w(j,e),w($,e),w(U,e),w(B,e),w(C,e),w(_,e),w(I,e),w(x,e),w(v,e),w(Z,e),w(X,e),w(H,e),w(Q,e),w(V,e),w(z,e),w(N,e),w(S,e),w(L,e),w(q,e),w(K,e),w(le,e),w(se,e),w(pe,e),w(ne,e),w(me,e),w(Me,e),w(oe,e),w(re,e),w(be,e),w(Je,e),w(he,e),w(_e,e),w(Ie,e),w(ke,e),w(xe,e),w(Ge,e)}}}const Pl='{"title":"JAX / Flaxμ—μ„œμ˜ 🧨 Stable Diffusion!","local":"jax--flaxμ—μ„œμ˜--stable-diffusion","sections":[{"title":"Setup","local":"setup","sections":[],"depth":2},{"title":"λͺ¨λΈ 뢈러였기","local":"λͺ¨λΈ-뢈러였기","sections":[],"depth":2},{"title":"μΆ”λ‘ ","local":"μΆ”λ‘ ","sections":[{"title":"볡사(Replication) 및 μ •λ ¬ν™”","local":"볡사replication-및-μ •λ ¬ν™”","sections":[],"depth":3},{"title":"μ‹œκ°ν™”","local":"μ‹œκ°ν™”","sections":[],"depth":3}],"depth":2},{"title":"λ‹€λ₯Έ ν”„λ‘¬ν”„νŠΈ μ‚¬μš©","local":"λ‹€λ₯Έ-ν”„λ‘¬ν”„νŠΈ-μ‚¬μš©","sections":[],"depth":2},{"title":"병렬화(parallelization)λŠ” μ–΄λ–»κ²Œ μž‘λ™ν•˜λŠ”κ°€?","local":"병렬화parallelizationλŠ”-μ–΄λ–»κ²Œ-μž‘λ™ν•˜λŠ”κ°€","sections":[],"depth":2}],"depth":1}';function El(tl){return Ql(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class ts extends Vl{constructor(u){super(),zl(this,u,El,Al,Fl,{})}}export{ts as component};

Xet Storage Details

Size:
36.1 kB
Β·
Xet hash:
003b5431ad481e5fac66d7fca15c8ee727e79b08357b9d573679a5ece2b1d70b

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.