-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathgenerate_dialogues_vllm.sh
More file actions
executable file
·29 lines (23 loc) · 1.11 KB
/
generate_dialogues_vllm.sh
File metadata and controls
executable file
·29 lines (23 loc) · 1.11 KB
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
#!/bin/bash
model_owner=$1
model=$2
tensor_parallel_size=$3
original_dataset=ATOMIC10X_persona_1k_3
langs=(chinese english french german portuguese spanish)
user_owner=google
user=gemma-3-27b-it
for lang in "${langs[@]}"; do
# Generate and validate turn 0 user starters
./0_generate_starters_vllm.sh ${user_owner} ${user} ${lang} ${tensor_parallel_size} ${original_dataset}
# Generate turn 0 assistant responses
./1_generate_turn0_vllm.sh ${model_owner} ${model} ${lang} ${user} ${tensor_parallel_size} ${original_dataset}
# Generate subsequent turns with validation for user responses
for turn in {1..4}; do
echo "Generating turn ${turn} for ${lang} - user role with validation"
# Generate and validate user responses
./4_generate_turnX.sh ${user_owner} ${user} ${lang} ${turn} ${model} ${tensor_parallel_size} user
echo "Generating turn ${turn} for ${lang} - assistant role"
# Generate assistant responses (no validation needed)
./4_generate_turnX.sh ${model_owner} ${model} ${lang} ${turn} ${user} ${tensor_parallel_size} assistant
done
done