AI Briefing

๐Ÿค— Transformers ์ง€์› ์–‘์žํ™” ๋ฐฉ์‹ ๊ฐ€์ด๋“œ

Overview of natively supported quantization schemes in ๐Ÿค— Transformers

ยท2023.09.12 09:00

๐Ÿค— Transformers ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ์—์„œ ๊ธฐ๋ณธ ์ง€์›ํ•˜๋Š” bitsandbytes์™€ auto-gptq ์–‘์žํ™” ๋ฐฉ์‹์˜ ํŠน์ง•์„ ๋น„๊ตํ•œ๋‹ค.

์–‘์žํ™”๋Š” ๋Œ€๊ทœ๋ชจ ๋ชจ๋ธ์„ ์†Œํ˜• ๊ธฐ๊ธฐ์—์„œ ์ถ”๋ก ํ•˜๊ฑฐ๋‚˜, ์–‘์žํ™”๋œ ๋ชจ๋ธ ์œ„์— **์–ด๋Œ‘ํ„ฐ๋ฅผ ๋ฏธ์„ธ ์กฐ์ •(Fine-tuning)**ํ•˜๊ธฐ ์œ„ํ•ด ์‚ฌ์šฉ๋œ๋‹ค.

ํ˜„์žฌ ๐Ÿค— Transformers์—์„œ ๋„ค์ดํ‹ฐ๋ธŒ๋กœ ์ง€์›ํ•˜๋Š” ์ฃผ์š” ์–‘์žํ™” ๋ฐฉ์‹์€ ๋‹ค์Œ๊ณผ ๊ฐ™๋‹ค:

  • bitsandbytes: ๋ชจ๋ธ ์ถ”๋ก  ๋ฐ ํšจ์œจ์ ์ธ ๋ฏธ์„ธ ์กฐ์ •์„ ์ง€์›ํ•œ๋‹ค.
  • auto-gptq: ๋ชจ๋ธ์˜ ์ถ”๋ก  ์†๋„ ์ตœ์ ํ™”์— ์ค‘์ ์„ ๋‘”๋‹ค.

์ฐธ๊ณ ๋กœ ๐Ÿค— Optimum ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ์—์„œ๋„ ๋‹ค์–‘ํ•œ ์–‘์žํ™” ๋ฐฉ์‹์„ ์ง€์›ํ•˜์ง€๋งŒ, ๋ณธ ๋‚ด์šฉ์€ Transformers ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ ๋‚ด์˜ ๊ธฐ๋ณธ ์ง€์› ๊ธฐ๋Šฅ์— ์ดˆ์ ์„ ๋งž์ถ˜๋‹ค. ๋˜ํ•œ, ํ˜„์žฌ ์ด ์ •๋ณด๋Š” PyTorch ๋ชจ๋ธ์—๋งŒ ์ ์šฉ๋˜๋ฉฐ TensorFlow ๋ฐ Flax/JAX ๋ชจ๋ธ์€ ์ง€์› ๋ฒ”์œ„์—์„œ ์ œ์™ธ๋œ๋‹ค.

์ด ์š”์•ฝ์€ ์›๋ฌธ ์ดํ•ด๋ฅผ ๋•๊ธฐ ์œ„ํ•œ ํ๋ ˆ์ด์…˜์ž…๋‹ˆ๋‹ค. ์ €์ž‘๊ถŒ์€ ์›์ €์ž‘์ž์—๊ฒŒ ์žˆ์œผ๋ฉฐ, ์ •ํ™•ํ•œ ๋‚ด์šฉ๊ณผ ๋งฅ๋ฝ์€ ์›๋ฌธ์„ ํ™•์ธํ•˜์„ธ์š”.

์š”์•ฝ ์˜ค๋ฅ˜, ์ถœ์ฒ˜ ํ‘œ๊ธฐ ๋ฌธ์ œ, ์‚ญ์ œ ์š”์ฒญ์€ ๋ฌธ์˜ ยท ๊ฑด์˜๋กœ ์•Œ๋ ค์ฃผ์„ธ์š”.