提交代码

models/resource_specification.go

@@ -285,6 +285,15 @@ func ResourceSpecOffShelf(id int64) (int64, error) {
 	return n, err
 }
 
+func GetResourceSpecificationByIds(ids []int64) ([]*Specification, error) {
+	r := make([]*Specification, 0)
+	err := x.In("id", ids).
+		Join("INNER", "resource_queue", "resource_queue.id = resource_specification.queue_id").
+		Find(&r)
+	return r, err
+
+}
+
 func GetResourceSpecification(r *ResourceSpecification) (*ResourceSpecification, error) {
 	has, err := x.Get(r)
 	if err != nil {

modules/setting/setting.go

@@ -713,14 +713,18 @@ var (
 	}{}
 
 	FileNoteBook = struct {
-		ProjectName  string
-		ImageGPU     string
-		SpecIdGPU    int64
-		SpecIdCPU    int64
-		ImageIdNPU   string
-		SpecIdNPU    int64
-		ImageIdNPUCD string
-		SpecIdNPUCD  int64
+		ProjectName           string
+		ImageGPU              string
+		SpecIdGPU             int64
+		SpecIdCPU             int64
+		ImageIdNPU            string
+		SpecIdNPU             int64
+		ImageIdNPUCD          string
+		SpecIdNPUCD           int64
+		ImageCPUDescription   string
+		ImageGPUDescription   string
+		ImageNPUDescription   string
+		ImageNPUCDDescription string
 	}{}
 
 	ModelConvert = struct {
@@ -1597,6 +1601,10 @@ func NewContext() {
 	FileNoteBook.SpecIdNPU = sec.Key("specid_npu").MustInt64(-1)
 	FileNoteBook.ImageIdNPUCD = sec.Key("imageid_npu_cd").MustString("")
 	FileNoteBook.SpecIdNPUCD = sec.Key("specid_npu_cd").MustInt64(-1)
+	FileNoteBook.ImageCPUDescription = sec.Key("image_cpu_desc").MustString("")
+	FileNoteBook.ImageGPUDescription = sec.Key("image_gpu_desc").MustString("")
+	FileNoteBook.ImageNPUDescription = sec.Key("image_npu_desc").MustString("")
+	FileNoteBook.ImageNPUCDDescription = sec.Key("image_npu_cd_desc").MustString("")
 
 	getGrampusConfig()
 	getModelartsCDConfig()