mirror of
https://github.com/hanzoai/ROCm.git
synced 2026-09-30 20:37:25 +00:00
58 lines
1.7 KiB
YAML
58 lines
1.7 KiB
YAML
pytorch_inference_benchmark:
|
|
unified_docker:
|
|
latest:
|
|
pull_tag: rocm/pytorch:latest
|
|
docker_hub_url:
|
|
rocm_version:
|
|
pytorch_version:
|
|
hipblaslt_version:
|
|
model_groups:
|
|
- group: CLIP
|
|
tag: clip
|
|
models:
|
|
- model: CLIP
|
|
mad_tag: pyt_clip_inference
|
|
model_repo: laion/CLIP-ViT-B-32-laion2B-s34B-b79K
|
|
url: https://huggingface.co/laion/CLIP-ViT-B-32-laion2B-s34B-b79K
|
|
precision: float16
|
|
- group: Chai-1
|
|
tag: chai
|
|
models:
|
|
- model: Chai-1
|
|
mad_tag: pyt_chai1_inference
|
|
model_repo: meta-llama/Llama-3.1-8B-Instruct
|
|
url: https://huggingface.co/chaidiscovery/chai-1
|
|
precision: float16
|
|
- group: Mochi Video
|
|
tag: mochi
|
|
models:
|
|
- model: Mochi 1
|
|
mad_tag: pyt_mochi_video_inference
|
|
model_repo: genmo/mochi-1-preview
|
|
url: https://huggingface.co/genmo/mochi-1-preview
|
|
precision: float16
|
|
- group: Wan2.1
|
|
tag: wan
|
|
models:
|
|
- model: Wan2.1
|
|
mad_tag: pyt_wan2.1_inference
|
|
model_repo: Wan-AI/Wan2.1-T2V-14B
|
|
url: https://huggingface.co/Wan-AI/Wan2.1-T2V-14B
|
|
precision: bfloat16
|
|
- group: Janus Pro
|
|
tag: janus-pro
|
|
models:
|
|
- model: Janus Pro 7B
|
|
mad_tag: pyt_janus_pro_inference
|
|
model_repo: deepseek-ai/Janus-Pro-7B
|
|
url: https://huggingface.co/deepseek-ai/Janus-Pro-7B
|
|
precision: bfloat16
|
|
- group: Hunyuan Video
|
|
tag: hunyuan
|
|
models:
|
|
- model: Hunyuan Video
|
|
mad_tag: pyt_hy_video
|
|
model_repo: tencent/HunyuanVideo
|
|
url: https://huggingface.co/tencent/HunyuanVideo
|
|
precision: float16
|