Init

Author: ring-c

main.go

@@ -1,19 +1,18 @@
 package main
 
 import (
-	"io"
 	"os"
 
 	"github.com/ring-c/go-web-diff/pkg/opts"
 	"github.com/ring-c/go-web-diff/pkg/sd"
 )
 
 func main() {
-
 	options := sd.DefaultOptions
 	options.GpuEnable = true
 	options.Wtype = opts.F16
 	options.Schedule = opts.KARRAS
+	options.Debug = true
 
 	model, err := sd.NewModel(options)
 	if err != nil {
@@ -30,56 +29,74 @@ func main() {
 		return
 	}
 
-	filenames := []string{
-		"./output/0.png",
+	// println(model.GetSystemInfo())
+
+	err = generate(model)
+	if err != nil {
+		println(err.Error())
+		return
 	}
 
-	var writers []io.Writer
-	for _, filename := range filenames {
-		var file *os.File
-		file, err = os.Create(filename)
-		if err != nil {
-			println(err.Error())
-			return
-		}
-
-		writers = append(writers, file)
-		defer func() {
-			_ = file.Close()
-		}()
+	// err = upscale(model)
+	// if err != nil {
+	// 	println(err.Error())
+	// 	return
+	// }
+}
+
+func generate(model *sd.Model) (err error) {
+	var file *os.File
+	file, err = os.Create("./output/0.png")
+	if err != nil {
+		println(err.Error())
+		return
 	}
 
+	defer func() {
+		_ = file.Close()
+	}()
+
 	var params = sd.DefaultFullParams
 	params.Width = 1024
 	params.Height = 1024
 	params.CfgScale = 2
-	params.SampleSteps = 32
+	params.SampleSteps = 4
 	params.SampleMethod = opts.EULER_A
-	params.BatchCount = len(writers)
+	params.Seed = 4242
 
-	err = model.Predict("1wolf, forest, full body", params, writers)
+	err = model.Predict("1girl, indoors, full body", params, file)
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+/*
+func upscale(model *sd.Model) (err error) {
+	fileRead, err := os.Open("./output/0.png")
+	if err != nil {
+		return
+	}
+
+	defer func() {
+		_ = fileRead.Close()
+	}()
+
+	fileWrite, err := os.Create("./output/1.png")
+	if err != nil {
+		return
+	}
+
+	defer func() {
+		_ = fileWrite.Close()
+	}()
+
+	err = model.UpscaleImage(fileRead, "/media/ed/files/sd/models/ESRGAN/RealESRGAN_x4plus_anime_6B.pth", 2, fileWrite)
 	if err != nil {
-		println(err.Error())
 		return
 	}
 
-	/*
-		fileRead, err := os.Open("./output/1.png")
-		if err != nil {
-			println(err.Error())
-			return
-		}
-
-		fileWrite, err := os.Create("./output/2.png")
-		if err != nil {
-			println(err.Error())
-			return
-		}
-
-		err = model.UpscaleImage(fileRead, "/media/ed/files/sd/models/ESRGAN/RealESRGAN_x4plus_anime_6B.pth", 2, fileWrite)
-		if err != nil {
-			println(err.Error())
-			return
-		}
-	*/
+	return
 }
+*/