sparse / ms-swift /examples /infer /pt /reward_model.sh

Upload folder using huggingface_hub

96fe658 verified about 1 month ago

181 Bytes

	CUDA_VISIBLE_DEVICES=0 \
	swift infer \
	--model Shanghai_AI_Laboratory/internlm2-1_8b-reward \
	--val_dataset AI-ModelScope/alpaca-gpt4-data-zh#1000 \
	--max_batch_size 64