dora/benches/llms/phi4.yaml

nodes:
  - id: benchmark_script
    build: |
      pip install ../mllm
    path: ../mllm/benchmark_script.py
    inputs:
      text: llm/text
    outputs:
      - text
    env:
      TEXT: "Please only generate the following output: This is a test"
      TEXT_TRUTH: "This is a test"

  - id: llm
    build: |
      pip install flash-attn --no-build-isolation
      pip install -e ../../node-hub/dora-phi4
    path: dora-phi4
    inputs:
      text: benchmark_script/text
    outputs:
      - text