From 9ce1b017b7bab8858f1ff45b2390b492ecfd56c4 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Tue, 13 Sep 2022 16:32:30 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/grampus/grampus.go |  6 +++---
 modules/setting/setting.go | 15 +++++++++------
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/modules/grampus/grampus.go b/modules/grampus/grampus.go
index 45c127141..3cdd59c5c 100755
--- a/modules/grampus/grampus.go
+++ b/modules/grampus/grampus.go
@@ -22,9 +22,6 @@ const (
 	GpuWorkDir = "/tmp/"
 	NpuWorkDir = "/cache/"
 
-	CommandPrepareScript = ";mkdir -p output;mkdir -p code;mkdir -p dataset;echo \"start loading script\";wget -q https://git.openi.org.cn/OpenIOSSG/script_for_grampus/archive/master.zip;" +
-		"echo \"finish loading script\";unzip -q master.zip;cd script_for_grampus;chmod 777 downloader_for_obs uploader_for_npu downloader_for_minio uploader_for_gpu;"
-
 	CodeArchiveName = "master.zip"
 )
 
@@ -34,6 +31,9 @@ var (
 	ImageInfos  *setting.StImageInfosModelArts
 
 	SpecialPools *models.SpecialPools
+
+	CommandPrepareScript = ";mkdir -p output;mkdir -p code;mkdir -p dataset;mkdir -p pretrainmodel;echo \"start loading script\";wget -q https://git.openi.org.cn/OpenIOSSG/" + setting.Grampus.SyncScriptProject + "/archive/master.zip;" +
+		"echo \"finish loading script\";unzip -q master.zip;cd script_for_grampus;chmod 777 downloader_for_obs uploader_for_npu downloader_for_minio uploader_for_gpu;"
 )
 
 type GenerateTrainJobReq struct {
diff --git a/modules/setting/setting.go b/modules/setting/setting.go
index 1992baf54..7d726a773 100755
--- a/modules/setting/setting.go
+++ b/modules/setting/setting.go
@@ -583,12 +583,13 @@ var (
 
 	//grampus config
 	Grampus = struct {
-		Env           string
-		Host          string
-		UserName      string
-		Password      string
-		SpecialPools  string
-		C2NetSequence string
+		Env               string
+		Host              string
+		UserName          string
+		Password          string
+		SpecialPools      string
+		C2NetSequence     string
+		SyncScriptProject string
 	}{}
 
 	C2NetInfos *C2NetSqInfos
@@ -1558,6 +1559,8 @@ func getGrampusConfig() {
 			log.Error("Unmarshal(C2NetSequence) failed:%v", err)
 		}
 	}
+	Grampus.SyncScriptProject = sec.Key("SYNC_SCRIPT_PROJECT").MustString("script_for_grampus")
+
 }
 
 func SetRadarMapConfig() {