导入模型支持智算网络。

Signed-off-by: zouap <zouap@pcl.ac.cn>

routers/repo/ai_model_manage.go

@@ -67,6 +67,7 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 			log.Info("download model from CloudBrainTwo faild." + err.Error())
 			return err
 		}
+		cloudType = models.TypeCloudBrainTwo
 	} else if aiTask.ComputeResource == models.GPUResource {
 		var ResourceSpecs *models.ResourceSpecs
 		json.Unmarshal([]byte(setting.ResourceSpecs), &ResourceSpecs)
@@ -81,6 +82,7 @@ func saveModelByParameters(jobId string, versionName string, name string, versio
 			log.Info("download model from CloudBrainOne faild." + err.Error())
 			return err
 		}
+		cloudType = models.TypeCloudBrainOne
 	}
 	//  else if cloudType == models.TypeC2Net {
 	// 	if aiTask.ComputeResource == models.NPUResource {