mirror of
https://github.com/ROCm/ROCm.git
synced 2026-01-09 22:58:17 -05:00
33 lines
1.0 KiB
YAML
33 lines
1.0 KiB
YAML
docker:
|
|
pull_tag: rocm/primus:v25.10
|
|
docker_hub_url: https://hub.docker.com/layers/rocm/primus/v25.10/images/sha256-140c37cd2eeeb183759b9622543fc03cc210dc97cbfa18eeefdcbda84420c197
|
|
components:
|
|
ROCm: 7.1.0
|
|
PyTorch: 2.10.0.dev20251112+rocm7.1
|
|
Python: "3.10"
|
|
Transformer Engine: 2.4.0.dev0+32e2d1d4
|
|
Flash Attention: 2.8.3
|
|
hipBLASLt: 1.2.0-09ab7153e2
|
|
model_groups:
|
|
- group: Meta Llama
|
|
tag: llama
|
|
models:
|
|
- model: Llama 3.1 8B
|
|
mad_tag: primus_pyt_train_llama-3.1-8b
|
|
model_repo: Llama-3.1-8B
|
|
url: https://huggingface.co/meta-llama/Llama-3.1-8B
|
|
precision: BF16
|
|
- model: Llama 3.1 70B
|
|
mad_tag: primus_pyt_train_llama-3.1-70b
|
|
model_repo: Llama-3.1-70B
|
|
url: https://huggingface.co/meta-llama/Llama-3.1-70B
|
|
precision: BF16
|
|
- group: DeepSeek
|
|
tag: deepseek
|
|
models:
|
|
- model: DeepSeek V2 16B
|
|
mad_tag: primus_pyt_train_deepseek-v2
|
|
model_repo: DeepSeek-V2
|
|
url: https://huggingface.co/deepseek-ai/DeepSeek-V2
|
|
precision: BF16
|