Upload folder using huggingface_hub

394918c verified 4 months ago

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

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