Abhishek Varma
dee41453b5
[SD] Add ControlNet to img2img + fix bug for img2img scheduler
...
-- This commit adds ControlNet execution to img2img.
-- It restructures the addition of ControlNet variants.
-- It also fixes scheduler selecting bug for img2img pipeline.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-24 19:07:19 +05:30
Gaurav Shukla
16168ab6b3
[SD] Update need_vae_encode correctly
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-21 20:26:06 +05:30
Gaurav Shukla
4233218629
[SD] Reset args.img_path to None in txt2img to avoid vae_encode
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-21 18:46:15 +05:30
Daniel Garvey
4e92304b89
remove annoying accelerate warning ( #1056 )
...
disables usage of low_cpu_mem_usage=True in from_pretrained() calls.
Can be re-enabled by using flag --low_cpu_mem_usage
defaults to False to avoid spam as we don't include accelerate in our
requirements.txt
2023-02-20 14:46:26 -06:00
Phaneesh Barwaria
d98b2afbe9
img2img denoise strength ( #1040 )
2023-02-16 13:40:20 -08:00
Gaurav Shukla
0d44a3527b
[SD][web] Add strength UI for img2img
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-15 22:47:41 +05:30
Gaurav Shukla
2147b6a397
[SD] Move some common code to utility
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-15 22:47:41 +05:30
Gaurav Shukla
6b5b4ba27b
[SD] Add batch count in Image2Image
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-15 22:47:41 +05:30
Gaurav Shukla
67005bf57c
[SD] Update iree-vulkan-target-triple after device switch
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-15 22:47:41 +05:30
PhaneeshB
0430c741c6
add strength param
2023-02-15 20:59:03 +05:30
powderluv
bc7333f1e5
Remove forcing LLPC setting ( #1018 )
...
also fix logo paths
2023-02-14 17:09:03 -08:00
powderluv
55ae50f991
Update inpaint.py
2023-02-14 14:12:05 -08:00
powderluv
a590c331ef
Update img2img.py
2023-02-14 14:11:50 -08:00
powderluv
8c241b06cb
Update txt2img.py
2023-02-14 14:11:36 -08:00
Gaurav Shukla
dd2e482214
[SD] Fix multiple call to device check ( #1007 )
...
- Also makes the dark theme default.
- Fix custom_vae parameter in img2img.
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-13 11:57:52 -08:00
drumicube
0464c7e558
Add support for command arguments to the WebUi ( #1000 )
...
Co-authored-by: powderluv <powderluv@users.noreply.github.com >
2023-02-11 19:20:21 -08:00
Abhishek Varma
9c13f1e635
Add custom vae support using --custom_vae flag
...
-- This commit adds custom vae support to SD wherein the user can
point to a model's checkpoint file whose Vae needs to be plugged
into the main model.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-10 23:33:47 +05:30
Gaurav Shukla
24af983cfb
[SD] Fix input image type
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-10 23:27:52 +05:30
Gaurav Shukla
67842a7525
[SD] Fix parameters in img2img
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-10 22:03:33 +05:30
PhaneeshB
3159a6f3e1
add support for img1img
2023-02-10 21:29:02 +05:30
jinchen62
6582475955
Add SD inpainting
...
python apps/stable_diffusion/scripts/inpaint.py --prompt="prompt" --img_path=path/to/img --mask_path=path/to/mask --import_mlir --max_length=77 --hf_model_id="stabilityai/stable-diffusion-2-inpainting"
2023-02-10 15:33:20 +05:30
Gaurav Shukla
5af124c5a5
[SD] Add batch count in stable diffusion
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-07 23:26:46 +05:30
drumicube
d35f992ace
Bring back the --runs options for the cmd command and fix wrong seed/model reported in json, csv and png ( #962 )
2023-02-06 15:16:50 -06:00
Gaurav Shukla
00cf73f9b8
[SD] Merge model id dropdown and .ckpt dropdown ( #936 )
...
- use_tuned is set to False for custom checkpoints.
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-03 10:43:33 -08:00
Abhishek Varma
5a5f285493
[apps-SD] Prepone loading of vmfbs + restructure the SD pipeline
...
-- This commit prepones loading of vmfbs, if present, for all sub-models.
-- It also involves restructuring the SD pipeline to achieve the loading
of vmfbs smoothly and postpones processing of checkpoint files only when
required.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-03 20:21:24 +05:30
Gaurav Shukla
339738f8a3
[SD][web] Populate checkpoints as dropdown UI ( #918 )
...
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-02 13:59:50 -08:00
powderluv
9b90672f63
Fix LLPC env var ( #920 )
2023-02-02 11:45:08 -08:00
Gaurav Shukla
5c7deb3611
[SD] Fix output image location ( #917 )
...
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-02 09:50:37 -08:00
Abhishek Varma
7cfc0fa55b
[APPS-SD] Fix a few bugs and bring it up to speed with SD CLI ( #908 )
2023-02-02 07:12:01 -08:00
powderluv
e26a70aa4f
Drop old cli and webui ( #911 )
2023-02-01 13:13:46 -08:00
powderluv
1c64e40722
Add PyInstaller for apps/ ( #907 )
...
Build with pyinstaller.exe .\apps\stable_diffusion\web\shark_sd.spec
normal flow works. exe is missing a few json files
2023-02-01 06:04:49 -08:00
Evan Guan
8cafe56eb4
Added flags for metadata information. ( #894 )
2023-02-01 05:16:11 -08:00
Ean Garvey
a504903dd5
Fix formatting issues. ( #903 )
2023-02-01 09:12:45 +05:30
Gaurav Shukla
c124b76328
[SD] Reorganize the stable diffusion model. ( #806 )
...
The stable diffusion codebase has been reorganized to make it more
modular so that the same script can be used for web as well as cli,
instead of duplicating the whole codebase.
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-01-31 14:42:41 -08:00