Update installer to download DreamShaper 8 model and sync tool checkpoint name

2026-02-22 23:58:58 +00:00
parent e49dcbd075
commit 8da1780f6e
2 changed files with 15 additions and 1 deletions
--- a/src/ollama_cli/tools/media.py
+++ b/src/ollama_cli/tools/media.py
@@ -30,7 +30,7 @@ def generate_image(prompt: str) -> str:
        # Based on the user's previous generate_image.py, but using the configured URL
        workflow = {
            "1": {
-                "inputs": {"ckpt_name": "dreamshaper_8.safetensors"},
+                "inputs": {"ckpt_name": "dreamshaper_8_pruned.safetensors"},
                "class_type": "CheckpointLoaderSimple"
            },
            "2": {