Commit Graph

156 Commits

Author SHA1 Message Date
JeLuF
7861c57317
Safetensor support (Fixes #599) (#608)
* safetensors support
Add support for checkpoints in safetensors format: https://github.com/huggingface/safetensors

This format shall be safer than pickle files

* pip install safetensors
2022-12-05 10:59:48 +05:30
cmdr2
cfd50231e1
Update on_sd_start.sh 2022-12-02 12:06:39 +05:30
cmdr2
54f7e6fcb8 SD2 fix - register buffer on the correct device 2022-11-30 16:05:06 +05:30
cmdr2
e37be0f954 Remove the need to use yield in the core loop for streaming results. This removes the need to patch the Stable Diffusion code, which can be fragile 2022-11-29 13:03:57 +05:30
cmdr2
13654cb8c0 Make on_sd_start.sh executable 2022-11-28 13:00:02 +05:30
cmdr2
642c114501 Working txt2img 2022-11-25 14:29:24 +05:30
cmdr2
02dd3e457d Tweaks to load sd1 models in sd2 code, typos 2022-11-25 13:57:15 +05:30
cmdr2
ea7b28c9d5 Placeholder changes for SD 2.0 support, haven't tested yet 2022-11-25 12:17:44 +05:30
cmdr2
8a0ec95fe1
Merge branch 'main' into beta 2022-11-23 15:08:34 +05:30
cmdr2
5a9c8e1d87 Warn but don't fix whitespaces in a patch 2022-11-22 23:21:11 +05:30
cmdr2
daaa65dc0a Warn but don't fix whitespaces in a patch 2022-11-22 23:20:24 +05:30
cmdr2
ab4e371524 Fix whitespace during git apply 2022-11-22 22:25:36 +05:30
cmdr2
5af84b8e90 Fix whitespace during git apply 2022-11-22 22:21:54 +05:30
cmdr2
d9b9f80a93 diffusion-kit upgrade 2022-11-22 17:39:51 +05:30
cmdr2
d429505b71 Update version of diffusion-kit 2022-11-22 17:14:20 +05:30
cmdr2
72ee708917 Remove the need to install realesrgan, gfpgan and certain specific package versions, since the new backend should install them directly 2022-11-22 16:50:10 +05:30
cmdr2
93bbfac29a Change the backend to a custom fork of SD, since basujindal's fork is no longer under development. This fork is intended to include the common models/tools used like RealESRGAN, GFPGAN, Codeformer etc, and is meant to be a community-developed project 2022-11-22 16:38:39 +05:30
cmdr2
7fc9509d4d Nowarn for fresh installation (git apply whitespace) 2022-11-21 17:20:09 +05:30
cmdr2
3bf5e11f94 Nowarn for fresh installation (git apply whitespace) 2022-11-21 17:19:55 +05:30
cmdr2
1cc27e524b Don't warn about whitespace in the git patch application 2022-11-21 17:11:57 +05:30
cmdr2
8316a002da Don't warn about whitespace in the git patch application 2022-11-21 17:11:38 +05:30
JeLuF
ed435d2b72
Add --whitespace=fix also on Linux 2022-11-20 23:09:27 +01:00
JeLuF
3e18f2f09c Picklescan of model files
During getModel(), the server picklescans the model files for potential
malicious code in the pickled python objects. If a malicious file is
found, the web UI will show a big red error message, the makeImage
button will be disabled, and the user must remove the malicious file
and reload the UI page.
2022-11-16 22:34:02 +01:00
cmdr2
e99d54d1f6 Merge main 2022-11-16 11:19:10 +05:30
cmdr2
9bc7521de0 Make custom VAE an Image Setting, rather than a System Setting; Don't load a VAE into memory by default 2022-11-08 16:54:15 +05:30
JeLuF
a62ee7850b Merge branch 'err-handle' of github.com:JeLuF/stable-diffusion-ui into err-handle 2022-11-05 23:31:19 +01:00
JeLuF
d3a90ccc0d Windows error handling
- cd to the script location on start of 'Start Stable Diffusion UI.cmd'
- Bail out when downloading micromamba fails
- add env variables SD_UI_BIND_IP and SD_UI_BIND_PORT to configure uvicorn
2022-11-05 23:30:40 +01:00
JeLuF
46b13ee664
Merge branch 'cmdr2:main' into err-handle 2022-11-05 20:49:19 +01:00
JeLuF
f969bfa7be More error handling, central error function 2022-11-05 18:59:59 +01:00
cmdr2
ea7006eec4 Set the PYTHONPATH before installation to prevent conda from knowing about any system-wide python installations (and skipping package installs due to that) 2022-11-02 19:45:49 +05:30
cmdr2
a8c16e39b8 Support custom VAE files; Use vae-ft-mse-840000-ema-pruned as the default VAE, which can be overridden by putting a .vae.pt file inside models/stable-diffusion with the same name as the ckpt model file. The UI / System Settings allows setting the default VAE model to use 2022-10-28 20:06:44 +05:30
cmdr2
6d43e0951c
Update on_sd_start.sh 2022-10-27 00:49:32 +05:30
cmdr2
088c546bee
Merge pull request #394 from JeLuF/patch-4
Retry downloads on flaky internet connections
2022-10-26 17:36:46 +05:30
JeLuF
4994a7ac85
Update on_sd_start.sh 2022-10-26 13:27:49 +02:00
cmdr2
c8fc0bb4f5 Remove the extra python version print 2022-10-26 16:02:00 +05:30
cmdr2
7b9d18caea Report the python version after activating the SD environment, that log is not very useful otherwise 2022-10-26 16:00:21 +05:30
cmdr2
137e519b66 Move the installer activation to on_sd_start 2022-10-26 12:15:58 +05:30
cmdr2
0c2d227da1 Activate the installer env while starting up, like the previous installer did in the start script 2022-10-25 19:21:26 +05:30
cmdr2
a281efef04 deactivate any pre-activated conda environments, by returning to (base) and then deactivating that. On Windows and Linux 2022-10-25 18:07:29 +05:30
cmdr2
538dcec348 deactivate any pre-activated conda environments before installing 2022-10-25 17:26:33 +05:30
cmdr2
0d38c8ae8f Bring back the post-installation test for torch 2022-10-25 16:38:08 +05:30
cmdr2
967c1a2da9 No need for conda 4.14, disable the broken test for torch 2022-10-25 15:42:22 +05:30
cmdr2
f3da326b77 Remove debug logging 2022-10-25 14:36:57 +05:30
cmdr2
153a6e2cb0 Temporarily disable sd_start for linux 2022-10-25 14:30:08 +05:30
cmdr2
95f37b9d36 Use conda 4.14 in the new installer as well (like the previous installer did) 2022-10-25 13:00:57 +05:30
cmdr2
60c37a1fc7 Set the PYTHONPATH before testing the SD installation 2022-10-25 12:48:56 +05:30
cmdr2
615c61e230 Fix a broken PYTHONPATH export in the linux script, used ; instead of : 2022-10-25 12:44:05 +05:30
cmdr2
a111d9b18a Supress detached head warning 2022-10-23 11:28:08 +05:30
cmdr2
cc356ce67d Initial commit of the new micromamba-based installer; This should work seamlessly for new and existing users; Also allows the installer to run on mac (but the installation will fail because the mac-specific environment.yaml hasn't been added yet) 2022-10-22 23:24:13 +05:30
cmdr2
e96312b470 Update the troubleshooting link to the new wiki page 2022-10-17 15:16:31 +05:30
cmdr2
06c8a004d8 Remove the old dev console scripts 2022-10-07 19:52:31 +05:30
cmdr2
4ab90fb14d Merge branch 'main' of github.com:cmdr2/stable-diffusion-ui 2022-10-07 19:48:40 +05:30
cmdr2
54d8feeaf0 Rename dev console script 2022-10-07 19:48:24 +05:30
cmdr2
10012d7125
Merge branch 'main' into beta 2022-10-07 19:38:53 +05:30
cmdr2
8c576ece28 Instructions for where to place the custom ckpt file 2022-10-07 19:33:52 +05:30
cmdr2
ab987e73c6 Copy the dev console script for linux on the first restart 2022-10-07 18:50:07 +05:30
cmdr2
201a053025 Support an arbitrary number of custom models, placed in the models/stable-diffusion folder. Shows an option in the UI to select which model to use 2022-10-06 14:28:02 +05:30
cmdr2
fcb41e30dc Fix a bug with the linux script, where the file size check would fail if the group name had spaces 2022-09-30 10:45:08 +05:30
cmdr2
1cdb456216 Fix a bug with the linux script, where the file size check would fail if the group name had spaces 2022-09-30 10:44:47 +05:30
cmdr2
8cb340be9d Merge branch 'main' into beta 2022-09-23 20:19:12 +05:30
cmdr2
b0c15bc430 Latest commit on basujindal's fork for the Linux version 2022-09-22 23:04:29 +05:30
cmdr2
1c5097b81b Hotfix for pywavelet version, attempt 1 2022-09-16 00:36:31 +05:30
cmdr2
2587727087 Merge branch 'main' of github.com:cmdr2/stable-diffusion-ui 2022-09-15 13:21:14 +05:30
cmdr2
4ccf26c23f Patch the openAI code only if it exists 2022-09-15 13:19:42 +05:30
cmdr2
476d6fe85d Force-install antlr4, since pip (incorrectly) skips installing it occasionally 2022-09-15 12:10:21 +05:30
cmdr2
e59c66ae26 Backend changes to support stopping a task mid-way. Uses a custom patch for the stable-diffusion codebase, to make it call a callback for DDIM 2022-09-13 19:59:41 +05:30
cmdr2
9ec2010ac2 Log python version in the linux build 2022-09-13 11:41:38 +05:30
cmdr2
812a0a14fc Export PYTHONPATH on linux 2022-09-13 10:36:50 +05:30
cmdr2
ba26f22f53 Compare file sizes 2022-09-12 21:44:16 +05:30
cmdr2
e1f37a2f3c Fix the comparison check for linux 2022-09-12 21:34:53 +05:30
cmdr2
e59287d736 Tweak the linux file size check code 2022-09-12 21:12:44 +05:30
cmdr2
a955730086 Allow using waifu 7 GB weights 2022-09-12 20:44:54 +05:30
cmdr2
a32a58bd0f Allow using the 7 GB model as well 2022-09-12 17:36:50 +05:30
cmdr2
ef46603f4e Tell conda to skip any pre-installed packages in the users' home/.local folder, since that can cause conflicts 2022-09-12 16:39:23 +05:30
cmdr2
905bcd8d1b Fix on linux 2022-09-10 00:37:04 +05:30
cmdr2
65b2c056c6 Revert "Revert "Revert "Revert "Merge pull request #112 from cmdr2/develop""""
This reverts commit 0dd38870e0.
2022-09-09 21:05:24 +05:30
cmdr2
53533e71e9 Revert "Revert "Revert "Revert "Merge main""""
This reverts commit 9d92174b1d.
2022-09-09 21:02:33 +05:30
cmdr2
b14523ecfa Re-enable updates for linux 2022-09-09 14:06:37 +05:30
cmdr2
9d92174b1d Revert "Revert "Revert "Merge main"""
This reverts commit a715022049.
2022-09-08 23:45:44 +05:30
cmdr2
0dd38870e0 Revert "Revert "Revert "Merge pull request #112 from cmdr2/develop"""
This reverts commit 788dcbf471.
2022-09-08 23:45:21 +05:30
cmdr2
a4305540f0 use a certain working version of SD 2022-09-08 23:23:02 +05:30
cmdr2
788dcbf471 Revert "Revert "Merge pull request #112 from cmdr2/develop""
This reverts commit 9051bf6e68.
2022-09-08 23:19:35 +05:30
cmdr2
a715022049 Revert "Revert "Merge main""
This reverts commit d92fb1ec95.
2022-09-08 23:19:20 +05:30
cmdr2
5170f508f7 Revert "Emergency fix"
This reverts commit 72900eaf93.
2022-09-08 23:18:50 +05:30
cmdr2
72900eaf93 Emergency fix 2022-09-08 23:12:32 +05:30
cmdr2
d92fb1ec95 Revert "Merge main"
This reverts commit ff515f9bb0, reversing
changes made to 10ed23e144.
2022-09-08 22:55:12 +05:30
cmdr2
9051bf6e68 Revert "Merge pull request #112 from cmdr2/develop"
This reverts commit 598de3697d, reversing
changes made to 0eae17075f.
2022-09-08 22:54:54 +05:30
cmdr2
ff515f9bb0 Merge main 2022-09-08 21:22:47 +05:30
cmdr2
253e75c747 v2.1 - Face correction (GFPGAN) and Upscaling (RealESRGAN) 2022-09-08 21:20:27 +05:30
elwynelwyn
8d596c07df Add shebangs to all sh files 2022-09-08 19:53:06 +12:00
cmdr2
3224cd73ed can't overwrite self - linux install script 2022-09-07 20:01:39 +05:30
cmdr2
a1908de302 Also check for antlr4 during the post-install tests 2022-09-07 15:49:14 +05:30
cmdr2
c30678af98 Fix the link for the troubleshooting page in all the scripts 2022-09-07 15:35:19 +05:30
cmdr2
c4349951da Create a troubleshooting page; Clearer troubleshooting steps in the installation error messages 2022-09-07 14:44:01 +05:30
cmdr2
0d9c8a804d Check the exact file size for the model file before allowing the installer to continue 2022-09-07 11:51:12 +05:30
cmdr2
0cc540b12a Incorrect file size check in linux 2022-09-06 19:13:03 +05:30
cmdr2
a1712a654d Add post-installation tests for the linux install script 2022-09-06 19:01:21 +05:30
cmdr2
7d69f4d3ed Fix #58 - While a .. in the path shouldn't cause any problems, just avoiding it entirely 2022-09-05 15:30:04 +05:30
cmdr2
d9a6e41265 Switch to a standardized model link, for v2 public release 2022-09-04 19:15:42 +05:30
cmdr2
e79e425cf5 Typo in linux v2 script for env var 2022-09-03 18:48:11 +05:30