change base model to stablediffusionapi/juggernaut-xl-v8

GrandlineAI · Jan 23, 2024 · b85316e · b85316e
1 parent 4c64b28
commit b85316e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -161,3 +161,8 @@ cython_debug/
 
 # Cog
 .cog
+checkpoints/
+models/
+output.jpg
+result.jpg
+
diff --git a/cog/predict.py b/cog/predict.py
@@ -104,7 +104,7 @@ def setup(self) -> None:
             local_files_only=True,
         )
 
-        base_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
+        base_model_path = "stablediffusionapi/juggernaut-xl-v8" #https://civitai.com/models/133005/juggernaut-xl
         self.pipe = StableDiffusionXLInstantIDPipeline.from_pretrained(
             base_model_path,
             controlnet=self.controlnet,