ConsumerBench/configs/workflow_content_creation.yml at main · efeslab/ConsumerBench · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
Brainstorm (chatbot):
    model: models/Llama-3.2-3B-Instruct-GGUF/Llama-3.2-3B-Instruct-f16.gguf
    num_requests: 10
    device: gpu
    type: Chatbot
    mps: 100

Analysis (deep_research):
    server_model: models/Llama-3.2-3B-Instruct-GGUF/Llama-3.2-3B-Instruct-f16.gguf
    client_model: openai/meta-llama/Llama-3.2-3B-Instruct
    num_requests: 1
    device: gpu
    type: DeepResearch
    mps: 100

Preparing Outline (chatbot):
    model: models/Llama-3.2-3B-Instruct-GGUF/Llama-3.2-3B-Instruct-f16.gguf
    num_requests: 20
    device: gpu
    type: Chatbot
    mps: 100

Creating Cover Art (imagegen):
    model: /mnt/tmpfs/models/stable-diffusion-3.5-medium-turbo
    num_requests: 10
    device: gpu
    type: ImageGen
    mps: 100

Generating Captions (live_captions):
    num_requests: 1
    device: gpu
    client_command_file: datasets/whisper-earnings21/5-min-chunks/4320211_chunk_002.wav
    mps: 100
    type: LiveCaptions

workflows:
    analysis:
        uses: Analysis (deep_research)
        background: true

    brainstorm:
        uses: Brainstorm (chatbot)

    outline:
        uses: Preparing Outline (chatbot)
        depend_on: ["brainstorm", "analysis"]

    cover_art:
        uses: Creating Cover Art (imagegen)
        depend_on: ["outline"]

    generate_captions:
        uses: Generating Captions (live_captions)
        depend_on: ["outline"]