Abhishek Varma
591bbcd058
[SD] Fix vmfb locating bug
...
-- This commit fixes a bug in vmfb caching due to vae_encoder and also
involves a minor NFC change in the code.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-10 23:33:47 +05:30
Abhishek Varma
99aa77d036
[SD] Add a common way to name vmfbs including custom_vae
...
-- This commit adds a common way to name vmfbs and adds to it `custom_vae`
support as well.
-- This was required to make a common place to change vmfbs name
without breaking any feature support AND also tackle the caching
of vmfbs gracefully.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-10 23:33:47 +05:30
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
Gaurav Shukla
b2f3c96835
[SD][web] Add Img2Img UI
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-02-10 21:27:31 +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
Anush Elangovan
41ee65b377
Revert "Enable --device_allocator=caching"
...
This reverts commit 83fe477066
2023-02-09 23:00:06 -08:00
Anush Elangovan
83fe477066
Enable --device_allocator=caching
2023-02-09 22:58:46 -08:00
yzhang93
4ca84ee4ee
Revert "Delete unnecessary arg setting ( #978 )" ( #985 )
...
This reverts commit 83c69ecd49
2023-02-09 16:44:26 -08:00
Ean Garvey
c28cc4c919
Fix local_tank_cache handling in shark_downloader. ( #981 )
2023-02-09 14:52:03 -06:00
yzhang93
e9864cb3f7
Modify the annotation OTF to return bytecode module ( #980 )
2023-02-08 14:29:43 -08:00
yzhang93
83c69ecd49
Delete unnecessary arg setting ( #978 )
2023-02-08 10:30:18 -08:00
Prashant Kumar
3595b4aaff
Incorporate latest changes in the shark_dynamo backend.
2023-02-08 20:37:30 +05:30
Abhishek Varma
3a9cfe113a
Fix SD restart error in exe file ( #975 )
...
-- This commit fixes SD restart error in exe file by creating
variants.json in CWD instead of a relative path.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
Co-authored-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-08 06:14:08 -08:00
yzhang93
c9966127da
Fix iree flags to be able to run on rdna2 ( #972 )
2023-02-07 16:39:32 -08:00
Ean Garvey
51300d33a7
Remove non-SD args from generate_sharktank.py ( #970 )
2023-02-07 13:29:55 -06:00
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
Abhishek Varma
eeb20b531a
Fix restart SD session error + override args.use_tuned temporarily
...
-- This commit fixes the session restart error for SD.
-- It also overrides `args.use_tuned` for `import_mlir`, and sets
`use_tuned` as `False`.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
2023-02-07 19:50:48 +05:30
cstueckrath
9dca842c22
Update .gitignore to exclude models ( #967 )
...
the models folder will be stashed along with other changes and most likely kill git doing so.
2023-02-07 01:48:36 -08:00
Ean Garvey
1eb9436836
Fix generate_sharktank args.
2023-02-07 14:06:07 +05:30
Ean Garvey
9604d9ce81
make --update_tank update only if hash mismatch
2023-02-07 14:06:07 +05:30
Ean Garvey
481d0553d8
Remove unnecessary repro_dir / shark_tmp usage
2023-02-07 14:06:07 +05:30
powderluv
60035cd63a
Add css in exe ( #963 )
...
exe should now default to dark theme too
2023-02-06 15:26:08 -08:00
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
Daniel Garvey
157ae64f9d
print to stdout for test visibility ( #937 )
...
Co-authored-by: dan <dan@nod-labs.com >
2023-02-06 01:03:27 -08:00
powderluv
ffa17f6057
Update sd_dark_theme.css
2023-02-06 01:01:50 -08:00
drumicube
d695a43e32
Make the dark theme default while launching web server ( #954 )
2023-02-05 07:25:45 -08:00
powderluv
01f6b4e6f0
Update README.md
2023-02-04 23:40:13 -08:00
yzhang93
7cf31a6ae4
Fix iree-benchmark flag names ( #952 )
2023-02-04 22:24:18 -08:00
Quinn Dawkins
fbd6224b04
Revert "Revert pipelines ( #948 )" ( #951 )
...
This reverts commit 8115b26079
2023-02-04 22:44:08 -05:00
powderluv
8115b26079
Revert pipelines ( #948 )
...
* Revert "[SD] Modify the flags to use --iree-preprocessing-pass-pipeline (#914 )"
This reverts commit a783c089a9#944 )"
This reverts commit 1d38d49162
2023-02-04 07:09:51 -08:00
powderluv
820586ac68
Update README.md
2023-02-04 01:01:11 -08:00
powderluv
4a7441ed07
Update profiling_with_iree.md
2023-02-04 00:47:57 -08:00
powderluv
383741f284
Update stable_diffusion_amd.md
2023-02-04 00:40:47 -08:00
powderluv
2bbc4e0e9f
Update README.md
2023-02-04 00:35:40 -08:00
powderluv
a7237244b0
Send users to the .exe file first
2023-02-04 00:30:32 -08:00
yzhang93
1d38d49162
Fix iree flags due to the change in shark-runtime ( #944 )
2023-02-03 21:34:02 -08:00
yzhang93
a783c089a9
[SD] Modify the flags to use --iree-preprocessing-pass-pipeline ( #914 )
...
* [SD] Modify the flags to use --iree-preprocessing-pass-pipeline
* Fix flags in sd_annotation
2023-02-03 15:08:02 -08:00
powderluv
e7907dc532
Disable tuned models for sm_89 ( #943 )
...
Looks like tuning on A100 doesn't necessarily translate to 40xx.
2023-02-03 14:30:46 -08:00
powderluv
394413679d
Fix ckpt_dir ( #939 )
2023-02-03 12:54:19 -08:00
powderluv
37189f14cb
roll to 492
2023-02-03 11:59:18 -08:00
powderluv
0b1ee81901
Minor webui changes ( #938 )
2023-02-03 11:26:45 -08: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
powderluv
7f2ea454b6
revert /base variants as they are different ( #929 )
...
sd2_1base is different from VAE base (for older cards)
2023-02-03 01:27:32 -08:00
Daniel Garvey
7c14002118
Map 2_1 to 2_1_base ( #927 )
...
* fix broken paths for older models
* adds a mapping from sd_2_1 to sd_2_1_base
we only have models in models_db for 2_1_base.
now that diffusers is fixed we can actually generate
2_1 itself, but until we add support for that in the tank
we should fetch 2_1_base for no-import_mlir
---------
Co-authored-by: dan <dan@nod-labs.com >
2023-02-02 19:03:19 -08:00
powderluv
3e9554f0a1
roll to 487
2023-02-02 19:02:39 -08:00
Daniel Garvey
e11ffec544
fix broken paths for older models ( #926 )
...
Co-authored-by: dan <dan@nod-labs.com >
2023-02-02 15:48:19 -08:00
