petals

(โ˜… 10,505)

๐ŸŒธ LLMs์„ BitTorrent ๋ฐฉ์‹์œผ๋กœ ์ง‘์—์„œ ์‹คํ–‰ํ•˜์„ธ์š”. ์˜คํ”„๋กœ๋”ฉ๋ณด๋‹ค ์ตœ๋Œ€ 10๋ฐฐ ๋น ๋ฅธ ํŒŒ์ธํŠœ๋‹๊ณผ ์ถ”๋ก 

  • .gitignore
  • Dockerfile
  • LICENSE
  • pyproject.toml
  • README.md
  • setup.cfg

# ์„ค์น˜ ๊ฐ€์ด๋“œ

1. ์ฝ”๋“œ ๋‚ด๋ ค๋ฐ›๊ธฐ
git clone https://github.com/bigscience-workshop/petals

๊นƒํ—ˆ๋ธŒ์—์„œ ํ”„๋กœ์ ํŠธ ์ฝ”๋“œ ์ „์ฒด๋ฅผ ๋‚ด ์ปดํ“จํ„ฐ๋กœ ๋‚ด๋ ค๋ฐ›์Šต๋‹ˆ๋‹ค.

cd petals

๋ฐฉ๊ธˆ ๋‚ด๋ ค๋ฐ›์€ ํ”„๋กœ์ ํŠธ ํด๋” ์•ˆ์œผ๋กœ ์ด๋™ํ•ฉ๋‹ˆ๋‹ค.

2. ๊ณต์‹ ์„ค์น˜ ์Šคํฌ๋ฆฝํŠธ

์‰ฌ์›€ ์ถ”์ฒœ
์‚ฌ์ „ ์ค€๋น„๋ฌผ
  • Python 3 pip ๋ช…๋ น์–ด๋ฅผ ์“ฐ๋ ค๋ฉด Python์ด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.
  • Homebrew macOS/Linux์šฉ ํŒจํ‚ค์ง€ ๋งค๋‹ˆ์ €์ž…๋‹ˆ๋‹ค.
pip install git+https://github.com/bigscience-workshop/petals

PyPI์— ๋ฐฐํฌ๋œ ํŒจํ‚ค์ง€๋ฅผ ๋ฐ”๋กœ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค. ์†Œ์Šค ํด๋ก ์ด ํ•„์š” ์—†์Šต๋‹ˆ๋‹ค.

brew install python

Homebrew๋กœ ์ด๋ฏธ ๋นŒ๋“œ๋œ ํŒจํ‚ค์ง€๋ฅผ ๋ฐ”๋กœ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค. ์†Œ์Šค ๋นŒ๋“œ๊ฐ€ ํ•„์š” ์—†์Šต๋‹ˆ๋‹ค.

โœ… ์„ค์น˜ ํ›„ ์ƒˆ ํ„ฐ๋ฏธ๋„์„ ์—ด๊ณ , ํ”„๋กœ๊ทธ๋žจ์˜ ๋ฒ„์ „ ํ™•์ธ ๋ช…๋ น(์˜ˆ: --version)์œผ๋กœ ์ •์ƒ ์„ค์น˜๋๋Š”์ง€ ํ™•์ธํ•˜์„ธ์š”.

์ด ๋ ˆํฌ์˜ README์— ์ ํžŒ ์‹ค์ œ ๋ช…๋ น์–ด๋ฅผ ๊ทธ๋Œ€๋กœ ๊ฐ€์ ธ์™”์Šต๋‹ˆ๋‹ค.

3. Docker

์‰ฌ์›€
์‚ฌ์ „ ์ค€๋น„๋ฌผ
  • Git GitHub์—์„œ ํ”„๋กœ์ ํŠธ ์ฝ”๋“œ๋ฅผ ๋‚ด๋ ค๋ฐ›์œผ๋ ค๋ฉด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.
  • Docker Desktop ์ปจํ…Œ์ด๋„ˆ๋ฅผ ๋นŒ๋“œํ•˜๊ณ  ์‹คํ–‰ํ•˜๋ ค๋ฉด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค. ์„ค์น˜ ํ›„ ์‹คํ–‰ํ•ด์„œ ๋ฐฑ๊ทธ๋ผ์šด๋“œ์— ์ผœ๋‘์„ธ์š”.
sudo docker run -p 31330:31330 --ipc host --gpus all --volume petals-cache:/cache --rm \

๋นŒ๋“œ๋œ ์ด๋ฏธ์ง€๋ฅผ ์‹ค์ œ ์ปจํ…Œ์ด๋„ˆ๋กœ ์‹คํ–‰ํ•ฉ๋‹ˆ๋‹ค.

โœ… ํ„ฐ๋ฏธ๋„์— docker compose ps ๋ฅผ ์ž…๋ ฅํ•ด ์ปจํ…Œ์ด๋„ˆ๋“ค์ด Up ์ƒํƒœ์ธ์ง€ ํ™•์ธํ•˜์„ธ์š”. README์— ํฌํŠธ ๋ฒˆํ˜ธ๊ฐ€ ์ ํ˜€์žˆ๋‹ค๋ฉด ๋ธŒ๋ผ์šฐ์ €์—์„œ http://localhost:ํฌํŠธ๋ฒˆํ˜ธ ๋กœ ์ ‘์†ํ•ด๋ณด์„ธ์š”.

์ด ๋ ˆํฌ์˜ README์— ์ ํžŒ ์‹ค์ œ ๋ช…๋ น์–ด๋ฅผ ๊ทธ๋Œ€๋กœ ๊ฐ€์ ธ์™”์Šต๋‹ˆ๋‹ค.

4. Python

์‰ฌ์›€
์‚ฌ์ „ ์ค€๋น„๋ฌผ
  • Git GitHub์—์„œ ํ”„๋กœ์ ํŠธ ์ฝ”๋“œ๋ฅผ ๋‚ด๋ ค๋ฐ›์œผ๋ ค๋ฉด ํ•„์š”ํ•ฉ๋‹ˆ๋‹ค.
  • Python 3 ์„ค์น˜ ์‹œ 'Add Python to PATH' ์˜ต์…˜์„ ๊ผญ ์ฒดํฌํ•˜์„ธ์š” (Windows).
pip install git+https://github.com/bigscience-workshop/petals

PyPI์— ๋ฐฐํฌ๋œ ํŒจํ‚ค์ง€๋ฅผ ๋ฐ”๋กœ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค. ์†Œ์Šค ํด๋ก ์ด ํ•„์š” ์—†์Šต๋‹ˆ๋‹ค.

python -m petals.cli.run_server meta-llama/Meta-Llama-3.1-405B-Instruct

ํŒŒ์ด์ฌ ์Šคํฌ๋ฆฝํŠธ(๋˜๋Š” ๋ชจ๋“ˆ)๋ฅผ ์‹คํ–‰ํ•ฉ๋‹ˆ๋‹ค.

python -m petals.cli.run_server --port 31330 meta-llama/Meta-Llama-3.1-405B-Instruct

ํŒŒ์ด์ฌ ์Šคํฌ๋ฆฝํŠธ(๋˜๋Š” ๋ชจ๋“ˆ)๋ฅผ ์‹คํ–‰ํ•ฉ๋‹ˆ๋‹ค.

python3 -m pip install git+https://github.com/bigscience-workshop/petals

requirements.txt ๋“ฑ์— ๋ช…์‹œ๋œ ํŒŒ์ด์ฌ ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๋ฅผ ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค.

python3 -m petals.cli.run_server meta-llama/Meta-Llama-3.1-405B-Instruct

ํŒŒ์ด์ฌ ์Šคํฌ๋ฆฝํŠธ(๋˜๋Š” ๋ชจ๋“ˆ)๋ฅผ ์‹คํ–‰ํ•ฉ๋‹ˆ๋‹ค.

โœ… ์—๋Ÿฌ ๋ฉ”์‹œ์ง€ ์—†์ด ์‹คํ–‰๋˜๊ณ  ํ„ฐ๋ฏธ๋„์— ์•ˆ๋‚ด ๋ฌธ๊ตฌ๊ฐ€ ์ถœ๋ ฅ๋˜๋ฉด ์ •์ƒ์ž…๋‹ˆ๋‹ค.

์ด ๋ ˆํฌ์˜ README์— ์ ํžŒ ์‹ค์ œ ๋ช…๋ น์–ด๋ฅผ ๊ทธ๋Œ€๋กœ ๊ฐ€์ ธ์™”์Šต๋‹ˆ๋‹ค.

// repository documentation