Josh Yan 9 months ago
parent
commit
ca1fbc5789
1 changed files with 5 additions and 4 deletions
  1. 5 4
      server/routes.go

+ 5 - 4
server/routes.go

@@ -614,11 +614,12 @@ func (s *Server) CreateModelHandler(c *gin.Context) {
 		defer cancel()
 		defer cancel()
 
 
 		quantization := cmp.Or(r.Quantize, r.Quantization)
 		quantization := cmp.Or(r.Quantize, r.Quantization)
-		if err := CreateModel(ctx, name, filepath.Dir(r.Path), strings.ToUpper(quantization), f, fn); errors.Is(err, errBadTemplate) {
-			ch <- gin.H{"error": err.Error(), "status": http.StatusBadRequest}
-		} else if err != nil {
+		if err := CreateModel(ctx, name, filepath.Dir(r.Path), strings.ToUpper(quantization), f, fn); err != nil {
+			if errors.Is(err, errBadTemplate) {
+			  ch <- gin.H{"error": err.Error(), "status": http.StatusBadRequest}
+			}
 			ch <- gin.H{"error": err.Error()}
 			ch <- gin.H{"error": err.Error()}
-		}
+		  }
 	}()
 	}()
 
 
 	if r.Stream != nil && !*r.Stream {
 	if r.Stream != nil && !*r.Stream {