diff --git a/NSIS/sdui.nsi b/NSIS/sdui.nsi index dc6c3769..5361512f 100644 --- a/NSIS/sdui.nsi +++ b/NSIS/sdui.nsi @@ -237,8 +237,8 @@ Section "MainSection" SEC01 CreateDirectory "$SMPROGRAMS\Easy Diffusion" CreateShortCut "$SMPROGRAMS\Easy Diffusion\Easy Diffusion.lnk" "$INSTDIR\Start Stable Diffusion UI.cmd" "" "$INSTDIR\installer_files\cyborg_flower_girl.ico" - DetailPrint 'Downloading the Stable Diffusion 1.4 model...' - NScurl::http get "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt" "$INSTDIR\models\stable-diffusion\sd-v1-4.ckpt" /CANCEL /INSIST /END + DetailPrint 'Downloading the Stable Diffusion 1.5 model...' + NScurl::http get "https://github.com/easydiffusion/sdkit-test-data/releases/download/assets/sd-v1-5.safetensors" "$INSTDIR\models\stable-diffusion\sd-v1-5.safetensors" /CANCEL /INSIST /END DetailPrint 'Downloading the GFPGAN model...' NScurl::http get "https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth" "$INSTDIR\models\gfpgan\GFPGANv1.4.pth" /CANCEL /INSIST /END diff --git a/scripts/check_modules.py b/scripts/check_modules.py index ffe473df..719a3bf7 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -18,7 +18,7 @@ os_name = platform.system() modules_to_check = { "torch": ("1.11.0", "1.13.1", "2.0.0", "2.0.1"), "torchvision": ("0.12.0", "0.14.1", "0.15.1", "0.15.2"), - "sdkit": "2.0.3", + "sdkit": "2.0.4", "stable-diffusion-sdkit": "2.1.4", "rich": "12.6.0", "uvicorn": "0.19.0", diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py index e57584ca..6d19912e 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -37,7 +37,7 @@ MODEL_EXTENSIONS = { } DEFAULT_MODELS = { "stable-diffusion": [ - {"file_name": "sd-v1-4.ckpt", "model_id": "1.4"}, + {"file_name": "sd-v1-5.safetensors", "model_id": "1.5-pruned-emaonly-fp16"}, ], "gfpgan": [ {"file_name": "GFPGANv1.4.pth", "model_id": "1.4"}, @@ -305,7 +305,7 @@ def is_malicious_model(file_path): def getModels(scan_for_malicious: bool = True): models = { "options": { - "stable-diffusion": [{"sd-v1-4": "SD 1.4"}], + "stable-diffusion": [], "vae": [], "hypernetwork": [], "lora": [],