make step 2 work

This commit is contained in:
Jade Choghari
2025-12-09 16:53:01 +00:00
parent c8eee4ea16
commit 025c2b2831
3 changed files with 52 additions and 11 deletions
+3 -3
View File
@@ -10,9 +10,9 @@ MODEL="Qwen/Qwen3-VL-30B-A3B-Instruct"
OUTPUT_DIR="/fsx/jade_choghari/outputs/pgen_annotations"
BATCH_SIZE=1
TEMPERATURE=0.7
SAMPLE_INTERVAL=1.0 # Generate dialogue every 1 second (all episodes processed)
BATCH_SIZE=32
TEMPERATURE=0.9
SAMPLE_INTERVAL=5.0 # Generate dialogue every 1 second (all episodes processed)
# Run synthetic data generation (processes ALL episodes)
python examples/dataset/annotate_pgen.py \