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
