diff --git a/.idea/deployment.xml b/.idea/deployment.xml
new file mode 100644
index 0000000..50a6fa2
--- /dev/null
+++ b/.idea/deployment.xml
@@ -0,0 +1,15 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="PublishConfigData" autoUpload="Always" remoteFilesAllowedToDisappearOnAutoupload="false">
+    <serverData>
+      <paths name="192.168.8.16">
+        <serverdata>
+          <mappings>
+            <mapping deploy="/Documents/stable-diffusion" local="$PROJECT_DIR$" web="/" />
+          </mappings>
+        </serverdata>
+      </paths>
+    </serverData>
+    <option name="myAutoUpload" value="ALWAYS" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/dictionaries/cyberhan.xml b/.idea/dictionaries/cyberhan.xml
new file mode 100644
index 0000000..f4f240c
--- /dev/null
+++ b/.idea/dictionaries/cyberhan.xml
@@ -0,0 +1,7 @@
+<component name="ProjectDictionaryState">
+  <dictionary name="cyberhan">
+    <words>
+      <w>upscaler</w>
+    </words>
+  </dictionary>
+</component>
\ No newline at end of file
diff --git a/.idea/remote-targets.xml b/.idea/remote-targets.xml
new file mode 100644
index 0000000..eb0e160
--- /dev/null
+++ b/.idea/remote-targets.xml
@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="RemoteTargetsManager">
+    <targets>
+      <target name="cyberhan@192.168.8.168:22" type="ssh/sftp" uuid="d5ed3230-c8f5-4935-837f-f91a89490256">
+        <config>
+          <option name="projectRootOnTarget" value="/C:/Users/cyberhan/Documents/stable-diffusion" />
+          <option name="serverName" value="cyberhan@192.168.8.168:22 password" />
+        </config>
+        <ContributedStateBase type="GoLanguageRuntime">
+          <config>
+            <option name="goPath" value="/C:/Users/cyberhan/go" />
+            <option name="goRoot" value="/C:/Program Files/Go/bin/go.exe" />
+          </config>
+        </ContributedStateBase>
+        <ContributedStateBase type="GoLanguageRuntime">
+          <config />
+        </ContributedStateBase>
+      </target>
+    </targets>
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/sshConfigs.xml b/.idea/sshConfigs.xml
new file mode 100644
index 0000000..630b92d
--- /dev/null
+++ b/.idea/sshConfigs.xml
@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="SshConfigs">
+    <configs>
+      <sshConfig authType="PASSWORD" connectionConfig="{&quot;serverAliveInterval&quot;:300}" host="192.168.8.168" id="23e06eb2-4a43-4987-82d7-253fdd7c4c03" port="22" nameFormat="DESCRIPTIVE" username="cyberhan" useOpenSSHConfig="true" />
+    </configs>
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/stable-diffusion.iml b/.idea/stable-diffusion.iml
index 30fbf67..42ef4cb 100644
--- a/.idea/stable-diffusion.iml
+++ b/.idea/stable-diffusion.iml
@@ -5,7 +5,12 @@
       <configuration sdkName="Python 3.11 (stable-diffusion)" />
     </facet>
   </component>
-  <component name="Go" enabled="true" />
+  <component name="Go" enabled="true">
+    <buildTags>
+      <option name="os" value="windows" />
+      <option name="arch" value="amd64" />
+    </buildTags>
+  </component>
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$">
       <excludeFolder url="file://$MODULE_DIR$/venv" />
diff --git a/.idea/webServers.xml b/.idea/webServers.xml
new file mode 100644
index 0000000..2eda7cc
--- /dev/null
+++ b/.idea/webServers.xml
@@ -0,0 +1,14 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="WebServers">
+    <option name="servers">
+      <webServer id="77340b47-5e04-43c2-ac19-e1432558082f" name="192.168.8.16">
+        <fileTransfer rootFolder="/C:/Users/cyberhan" accessType="SFTP" host="192.168.8.168" port="22" sshConfigId="23e06eb2-4a43-4987-82d7-253fdd7c4c03" sshConfig="cyberhan@192.168.8.168:22 password">
+          <advancedOptions>
+            <advancedOptions dataProtectionLevel="Private" passiveMode="true" shareSSLContext="true" />
+          </advancedOptions>
+        </fileTransfer>
+      </webServer>
+    </option>
+  </component>
+</project>
\ No newline at end of file
diff --git a/sd.go b/sd.go
index 65010ed..0c56e3a 100644
--- a/sd.go
+++ b/sd.go
@@ -317,9 +317,10 @@ func imageToBytes(decode image.Image) Image {
 		}
 	}
 	return Image{
-		Width:  uint32(width),
-		Height: uint32(height),
-		Data:   bytesImg,
+		Width:   uint32(width),
+		Height:  uint32(height),
+		Data:    bytesImg,
+		Channel: 3,
 	}
 }
 
diff --git a/sd_test.go b/sd_test.go
index cddff27..373b7e5 100644
--- a/sd_test.go
+++ b/sd_test.go
@@ -60,6 +60,9 @@ func TestNewStableDiffusionAutoModelImagePredict(t *testing.T) {
 		return
 	}
 	defer model.Close()
+	model.SetLogCallback(func(level sd.LogLevel, msg string) {
+		t.Log(msg)
+	})
 	err = model.LoadFromFile("./models/mysd.safetensors")
 	if err != nil {
 		t.Error(err)
@@ -75,7 +78,7 @@ func TestNewStableDiffusionAutoModelImagePredict(t *testing.T) {
 	var writers []io.Writer
 	filenames := []string{
 		"./assets/love_cat0_m.png",
-		"./assets/love_cat1_m.png",
+		//"./assets/love_cat1_m.png",
 		//"./assets/love_cat5.png",
 		//"./assets/love_cat6.png"
 	}
@@ -88,8 +91,12 @@ func TestNewStableDiffusionAutoModelImagePredict(t *testing.T) {
 		defer file.Close()
 		writers = append(writers, file)
 	}
-
-	err = model.ImagePredict(inFile, "the cat that wears shoe", sd.DefaultFullParams, writers)
+	params := sd.DefaultFullParams
+	params.BatchCount = 1
+	params.Width = 256
+	params.Height = 256
+	params.NegativePrompt = ""
+	err = model.ImagePredict(inFile, "dogs", params, writers)
 	if err != nil {
 		t.Error(err)
 		return