提交代码。

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

routers/repo/ai_model_convert.go

@@ -99,10 +99,12 @@ func SaveModelConvert(ctx *context.Context) {
 		UserId:          ctx.User.ID,
 	}
 	models.SaveModelConvert(modelConvert)
-	if modelConvert.SrcEngine == PYTORCH_ENGINE {
+	if modelConvert.SrcEngine == PYTORCH_ENGINE || modelConvert.SrcEngine == TENSORFLOW_ENGINE {
+		log.Info("create gpu train job.")
 		err = createGpuTrainJob(modelConvert, ctx, task.Path)
 	} else {
 		//create npu job
+		log.Info("create npu train job.")
 		createNpuTrainJob(modelConvert, ctx, task.Path)
 	}
 
@@ -386,7 +388,7 @@ func DeleteModelConvert(ctx *context.Context) {
 }
 
 func isCloudBrainTask(task *models.AiModelConvert) bool {
-	if task.SrcEngine == PYTORCH_ENGINE {
+	if task.SrcEngine == PYTORCH_ENGINE || task.SrcEngine == TENSORFLOW_ENGINE {
 		return true
 	}
 	return false