sd 20 in twitch, use sd turbo original model for automatic11111

groovybits · Mar 27, 2024 · 1021fd1 · 1021fd1
1 parent 9c91aea
commit 1021fd1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/twitch.sh b/scripts/twitch.sh
@@ -41,7 +41,7 @@ SD_HEIGHT=512
 SD_API=1
 SD_MODEL=turbo
 SD_INTERMEDIARY_IMAGES=1
-SD_N_STEPS=25
+SD_N_STEPS=20
 ALIGNMENT=right
 SUBTITLES=1
 # === END OF CONFIGURATION ===

diff --git a/src/sd_automatic.rs b/src/sd_automatic.rs
@@ -18,7 +18,7 @@ pub async fn sd_auto(
         StableDiffusionVersion::V1_5 => "v1-5-pruned-emaonly.ckpt",
         StableDiffusionVersion::V2_1 => "v2-1_768-ema-pruned.ckpt",
         StableDiffusionVersion::Xl => "stabilityai/stable-diffusion-xl-1024-1.0.ckpt",
-        StableDiffusionVersion::Turbo => "stabilityai/sd_xl_turbo_1.0.safetensors",
+        StableDiffusionVersion::Turbo => "madebyollin/turbo-diffusion.ckpt",
     };
 
     let payload = AutomaticPayload {