浏览代码

models: rm workDir from CreateModel

unused after removing EMBED
Michael Yang 1 年之前
父节点
当前提交
a19d47642e
共有 2 个文件被更改,包括 2 次插入4 次删除
  1. 1 1
      server/images.go
  2. 1 3
      server/routes.go

+ 1 - 1
server/images.go

@@ -252,7 +252,7 @@ func filenameWithPath(path, f string) (string, error) {
 	return f, nil
 	return f, nil
 }
 }
 
 
-func CreateModel(ctx context.Context, workDir, name string, path string, fn func(resp api.ProgressResponse)) error {
+func CreateModel(ctx context.Context, name string, path string, fn func(resp api.ProgressResponse)) error {
 	mp := ParseModelPath(name)
 	mp := ParseModelPath(name)
 
 
 	var manifest *ManifestV2
 	var manifest *ManifestV2

+ 1 - 3
server/routes.go

@@ -334,8 +334,6 @@ func CreateModelHandler(c *gin.Context) {
 		return
 		return
 	}
 	}
 
 
-	workDir := c.GetString("workDir")
-
 	ch := make(chan any)
 	ch := make(chan any)
 	go func() {
 	go func() {
 		defer close(ch)
 		defer close(ch)
@@ -346,7 +344,7 @@ func CreateModelHandler(c *gin.Context) {
 		ctx, cancel := context.WithCancel(c.Request.Context())
 		ctx, cancel := context.WithCancel(c.Request.Context())
 		defer cancel()
 		defer cancel()
 
 
-		if err := CreateModel(ctx, workDir, req.Name, req.Path, fn); err != nil {
+		if err := CreateModel(ctx, req.Name, req.Path, fn); err != nil {
 			ch <- gin.H{"error": err.Error()}
 			ch <- gin.H{"error": err.Error()}
 		}
 		}
 	}()
 	}()