xfcghj
/

AR

Model card Files Files and versions

AR / scripts /test.sh

xfcghj's picture

Upload folder using huggingface_hub

394918c verified about 1 month ago

History Blame Contribute Delete

367 Bytes

	#!/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