mirror of
https://github.com/hanzoai/ROCm.git
synced 2026-10-30 14:49:41 +00:00
peterjunparkandGitHub
4d156fcbb3
* add previous versions * Fix heading levels in pages using embedded templates (#5468) * update primus-megatron doc update megatron-lm doc update templates fix tab update primus-megatron model configs Update primus-pytorch model configs fix css class add posttrain to pytorch-training template update data sheets update update update update docker tags * Add known issue and update Primus/Turbo versions * add primus ver to histories * update primus ver to 0.1.1 * fix leftovers from merge conflict
25 lines
844 B
YAML
25 lines
844 B
YAML
dockers:
|
|
- pull_tag: rocm/pytorch-training:v25.8
|
|
docker_hub_url: https://hub.docker.com/layers/rocm/pytorch-training/v25.8/images/sha256-5082ae01d73fec6972b0d84e5dad78c0926820dcf3c19f301d6c8eb892e573c5
|
|
components:
|
|
ROCm: 6.4.3
|
|
PyTorch: 2.8.0a0+gitd06a406
|
|
Python: 3.10.18
|
|
Transformer Engine: 2.2.0.dev0+a1e66aae
|
|
Flash Attention: 3.0.0.post1
|
|
hipBLASLt: 1.1.0-d1b517fc7a
|
|
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
|