From bcaad140910b1200d50743c663e4331910358aa6 Mon Sep 17 00:00:00 2001
From: FantasticGNU <60054847+FantasticGNU@users.noreply.github.com>
Date: Wed, 11 Oct 2023 15:44:12 +0800
Subject: [PATCH] Update train_mvtec.sh

---
 code/scripts/train_mvtec.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/code/scripts/train_mvtec.sh b/code/scripts/train_mvtec.sh
index ddf1b46..21cfcf5 100644
--- a/code/scripts/train_mvtec.sh
+++ b/code/scripts/train_mvtec.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-deepspeed --include localhost:5,7 --master_port 28400 train_mvtec.py \
+deepspeed --include localhost:0,1 --master_port 28400 train_mvtec.py \
     --model openllama_peft \
     --stage 1\
     --imagebind_ckpt_path ../pretrained_ckpt/imagebind_ckpt/imagebind_huge.pth\