File size: 367 Bytes
f0fc238 394918c f0fc238 394918c | 1 2 3 4 5 6 7 8 9 10 11 | #!/bin/bash
export CUDA_VISIBLE_DEVICES=1,2,3
python test.py \
--checkpoint_path /home/dataset-assist-0/usr/lh/ysh/dw/RL/AR/checkpoints/20260614_235422/model_epoch_7.pth \
--archive_path /home/dataset-assist-0/usr/lh/ysh/dw/RL/AR/data/shards/shard_0000.tar.zst \
--num_refine_groups 3 \
--model_device cuda:0 \
--vae_device cuda:0 \
--clip_device cuda:0 |