rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

SD Typo Fix

This commit is contained in:
Henk 2023-11-03 15:03:23 +01:00
parent aec6bcecb6
commit f43896750a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -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("mpeterwilli/deliberate-2", local_dir="functional_models/stable-diffusion", local_dir_use_symlinks=False, cache_dir="cache/", ignore_patterns=["*.safetensors"])
snapshot_download("peterwilli/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")