diff --git a/routers/repo/ai_model_convert.go b/routers/repo/ai_model_convert.go
index 85824ee01..87b11ecd6 100644
--- a/routers/repo/ai_model_convert.go
+++ b/routers/repo/ai_model_convert.go
@@ -99,7 +99,7 @@ func SaveModelConvert(ctx *context.Context) {
 		UserId:          ctx.User.ID,
 	}
 	models.SaveModelConvert(modelConvert)
-	if modelConvert.SrcEngine == PYTORCH_ENGINE || modelConvert.SrcEngine == TENSORFLOW_ENGINE {
+	if modelConvert.SrcEngine == PYTORCH_ENGINE {
 		err = createGpuTrainJob(modelConvert, ctx, task.Path)
 	} else {
 		//create npu job