Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
Jintao-Huang committed Dec 27, 2024
1 parent 5530e76 commit 928ae88
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion examples/train/multimodal/infer.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# Perform inference using the validation set from the training phase.
# CUDA_VISIBLE_DEVICES=0,1 \
CUDA_VISIBLE_DEVICES=0 \
swift infer \
--adapters output/vx-xxx/checkpoint-xxx \
--stream true \
Expand Down
6 changes: 3 additions & 3 deletions examples/train/multimodal/ocr.sh
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
# 2*75GB
CUDA_VISIBLE_DEVICES=0,1 \
# 20GB
CUDA_VISIBLE_DEVICES=0 \
MAX_PIXELS=1003520 \
swift sft \
--model Qwen/QVQ-72B-Preview \
--model Qwen/Qwen2-VL-7B-Instruct \
--dataset 'AI-ModelScope/LaTeX_OCR:human_handwrite#20000' \
--train_type lora \
--torch_dtype bfloat16 \
Expand Down

0 comments on commit 928ae88

Please sign in to comment.