grok-1 README

The project's README from github.com/xai-org/grok-1. As one markdown file: README.md.

Grok-1

This repository contains JAX example code for loading and running the Grok-1 open-weights model.

Make sure to download the checkpoint and place the ckpt-0 directory in checkpoints - see Downloading the weights

Then, run

pip install -r requirements.txt
python run.py

to test the code.

The script loads the checkpoint and samples from the model on a test input.

Due to the large size of the model (314B parameters), a machine with enough GPU memory is required to test the model with the example code. The implementation of the MoE layer in this repository is not efficient. The implementation was chosen to avoid the need for custom kernels to validate the correctness of the model.

Model Specifications

Grok-1 is currently designed with the following specifications:

Downloading the weights

You can download the weights using a torrent client and this magnet link:

magnet:?xt=urn:btih:5f96d43576e3d386c9ba65b883210a393b68210e&tr=https%3A%2F%2Facademictorrents.com%2Fannounce.php&tr=udp%3A%2F%2Ftracker.coppersurfer.tk%3A6969&tr=udp%3A%2F%2Ftracker.opentrackr.org%3A1337%2Fannounce

or directly using HuggingFace 🤗 Hub:

git clone https://github.com/xai-org/grok-1.git && cd grok-1
pip install huggingface_hub[hf_transfer]
huggingface-cli download xai-org/grok-1 --repo-type model --include ckpt-0/* --local-dir checkpoints --local-dir-use-symlinks False

License

The code and associated Grok-1 weights in this release are licensed under the Apache 2.0 license. The license only applies to the source files in this repository and the model weights of Grok-1.

grok-1 (xai-org/grok-1)

Searching Database…

52.2K ·GitHub
Что это такое?Grok-1 — открытая xAI языковая модель с 314 млрд параметров на архитектуре Mixture-of-Experts; в этом репозитории лежит пример кода на JAX для её загрузки и запуска.
Веса модели тут есть или только код?Только код — открытые веса скачиваются отдельно: их можно получить через торрент-клиент по магнит-ссылке или напрямую с HuggingFace Hub.
Как запустить, когда веса уже скачаны?Положите скачанный каталог ckpt-0 внутрь checkpoints, затем запустите тестовый скрипт: он загрузит чекпоинт и сгенерирует ответ модели на тестовый запрос.
Приложенный код эффективный?Нет — слой MoE намеренно сделан неэффективно: без кастомных ядер, чтобы корректность работы модели было легко проверить.
Лицензия Apache 2.0 покрывает и код, и веса?Да — и код, и веса Grok-1 распространяются под Apache 2.0: она действует и на исходники этого репозитория, и на веса модели Grok-1.
Это запустится на обычной видеокарте?Практически нет — при 314 млрд параметров README требует машину с достаточным объёмом видеопамяти, даже чтобы просто протестировать модель на примере кода.
Насколько большая эта модель?314 млрд параметров в архитектуре Mixture-of-8-Experts: на каждый токен используются 2 эксперта, 64 слоя, 48 голов внимания для запросов (8 для ключей/значений) и размер эмбеддинга 6 144.
Какой длины контекст он выдерживает?До 8 192 токенов контекста; текст разбивается на токены токенизатором SentencePiece со словарём на 131 072 токена.
Есть ли поддержка квантизации или шардирования?Да — модель поддерживает шардирование активаций и 8-битную квантизацию, а для позиционной информации использует ротационные эмбеддинги (RoPE).
Creator
xai-org
Size
1 MB
Stars
52.2K
Updated
CommitMar 19
Age
2y
Commits · 90 Days