From aec6bcecb676663bfaaf399c8dbd2f742361ef34 Mon Sep 17 00:00:00 2001 From: Henk Date: Fri, 3 Nov 2023 14:45:41 +0100 Subject: [PATCH] New SD URL --- aiserver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index c4b82b3a..6ce07d2a 100644 --- a/aiserver.py +++ b/aiserver.py @@ -7382,7 +7382,7 @@ def text2img_local(prompt: str) -> Optional[Image.Image]: if koboldai_vars.image_pipeline is None: if not os.path.exists("functional_models/stable-diffusion/model_index.json"): from huggingface_hub import snapshot_download - snapshot_download("XpucT/Deliberate", local_dir="functional_models/stable-diffusion", local_dir_use_symlinks=False, cache_dir="cache/", ignore_patterns=["*.safetensors"]) + snapshot_download("mpeterwilli/deliberate-2", local_dir="functional_models/stable-diffusion", local_dir_use_symlinks=False, cache_dir="cache/", ignore_patterns=["*.safetensors"]) pipe = tpool.execute(StableDiffusionPipeline.from_pretrained, "functional_models/stable-diffusion", safety_checker=None, torch_dtype=torch.float16).to("cuda") else: pipe = koboldai_vars.image_pipeline.to("cuda")