yzhang93
17a67897d1
Add SD v2.1 768x768 tuned model ( #1244 )
...
Co-authored-by: powderluv <powderluv@users.noreply.github.com >
20230326.640
20230325.639
20230324.638
20230324.637
2023-03-24 10:39:15 -07:00
Gaurav Shukla
da449b73aa
[SD] Disable lora training tab for now ( #1241 )
...
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-03-24 09:16:24 -07:00
Kyle Herndon
0b0526699a
Fix incorrect device argument initialization for LoRA training by extracting the device type and number and formatting it for pytorch ( #1237 )
...
Co-authored-by: Kyle Herndon <kyle@nod-labs.com >
2023-03-24 01:10:50 -07:00
Boian Petkantchin
4fac46f7bb
In models testing fix paths to be relative to the script dir not cwd ( #1128 )
...
authored-by: Boian Petkantchin <boian@nod-labs.com >
20230323.636
2023-03-22 15:26:52 -05:00
Daniel Garvey
49925950f1
fix false positives ( #1193 )
2023-03-22 15:25:39 -05:00
Thomas
807947c0c8
Remove deprecated cli option iree-hal-cuda-disable-loop-nounroll-wa ( #1235 )
2023-03-22 12:05:15 -05:00
Abhishek Varma
593428bda4
[SD] Fix for transformers/__init__.py issue in PyInstaller ( #1233 )
...
-- This commit fixes the transformers/__init__.py issue in PyInstaller.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
Co-authored-by: Abhishek Varma <abhishek@nod-labs.com >
2023-03-22 08:43:53 -07:00
Abhishek Varma
cede9b4fec
[SD] Fix custom_vae as a required parameter in inpaint ( #1232 )
2023-03-22 04:30:17 -07:00
Prashant Kumar
c2360303f0
Add the int8 quantized model.
2023-03-22 16:28:13 +05:30
jinchen62
420366c1b8
Move schedulers to global obj ( #1225 )
2023-03-21 22:40:43 -07:00
Ean Garvey
d31bae488c
Set iree-input-type to tm_tensor for SD ( #1228 )
20230321.628
20230321.627
2023-03-21 19:07:31 -07:00
Kyle Herndon
c23fcf3748
Fix incorrect device argument initialization for LoRA training ( #1231 )
...
Co-authored-by: Kyle Herndon <kyle@nod-labs.com >
Co-authored-by: powderluv <powderluv@users.noreply.github.com >
2023-03-21 19:07:18 -07:00
jinchen62
7dbbb1726a
Fix SD obj not defined if fail to get models from pretrained ( #1222 )
20230321.625
2023-03-21 07:55:17 -07:00
Abhishek Varma
8b8cc7fd33
[SD] Update LoRA inference to handle various checkpoints ( #1215 )
2023-03-21 06:52:20 -07:00
Ean Garvey
e3c96a2b9d
Move sentencepiece to importer requirements. ( #1218 )
20230320.624
2023-03-21 00:39:57 -05:00
Ean Garvey
5e3f50647d
Set --vulkan_large_heap_block_size default to 2gb. ( #1220 )
2023-03-20 21:07:09 -07:00
gpetters94
7899e1803a
Add fix for attention slicing fp16 ( #1217 )
2023-03-20 19:11:29 -07:00
mariecwhite
d105246b9c
Fix t5 models
2023-03-21 10:39:59 +11:00
mariecwhite
90c958bca2
Add T5-base and T5-large Torch and TF Models ( #1116 )
2023-03-20 17:32:50 -05:00
mariecwhite
f99903e023
Add EfficientNet B0 and B7 Torch and TF models
2023-03-21 09:22:05 +11:00
mariecwhite
c6f44ef1b3
Add EfficientNet B0 and B7 Torch and TF models
2023-03-21 09:14:45 +11:00
mariecwhite
8dcd4d5aeb
Make batch size configurable
2023-03-20 18:03:17 -04:00
Phoenix Meadowlark
d319f4684e
Add peak memory reporting for IREE, TF and PyTorch ( #1216 )
2023-03-20 15:40:49 -05:00
Ean Garvey
54d7b6d83e
Generate model artifacts in pytests if they don't exist in the cloud. ( #1121 )
...
* Add gen_shark_files fn to shark_downloader for OTF artifact generation
* add generate_sharktank as a tank/ python module.
* Fix some paths in tank generation.
2023-03-20 12:13:19 -05:00
m68k-fr
4a622532e5
[Web] Stop images ( #1212 )
20230319.619
20230319.618
20230319.617
2023-03-19 14:37:30 -07:00
cstueckrath
650b2ada58
add pytorch_lightning to requirements ( #1211 )
...
* add pytorch_lightning to requirements
this will additionally add lightning-utilities and torchmetrics
* Update shark_sd.spec
* Update shark_sd_cli.spec
20230319.616
20230319.615
2023-03-19 12:29:54 -07:00
m68k-fr
f87f8949f3
[Web] CSS fix for gradio V3.22.1 ( #1210 )
20230319.614
2023-03-19 06:13:59 -07:00
m68k-fr
7dc9bf8148
[Web] Move "stop Batch" button to "Advanced Options" toggle ( #1209 )
2023-03-18 20:54:42 -07:00
Kyle Herndon
ba48ff8d25
Implement LoRA training and UI for training and UI for inference in img2img, inpaint, outpaint ( #1200 )
...
txt2img inference UI is already committed.
Co-authored-by: Kyle Herndon <kyle@nod-labs.com >
Co-authored-by: powderluv <powderluv@users.noreply.github.com >
20230317.609
2023-03-17 12:54:56 -07:00
Gaurav Shukla
638840925c
[SD] Add support for larger size upscaling ( #1204 )
...
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-03-17 10:20:48 -07:00
m68k-fr
b661656c03
[Web] Fix custom model path for upscaler ( #1199 )
20230316.607
20230316.606
2023-03-16 15:57:23 -07:00
Gaurav Shukla
0225434389
[SD] Add sendTo Upscaler
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-03-16 20:49:19 +05:30
Gaurav Shukla
7ffe20b1c2
[SD] Release memory used by upscaler when not in use
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-03-16 20:49:19 +05:30
Gaurav Shukla
d8f0c4655d
[SD] Add Upscaler web
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-03-16 20:49:19 +05:30
Gaurav Shukla
7e8d3ec0df
[SD] Add upscalar pipeline
...
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com >
2023-03-16 20:49:19 +05:30
jinchen62
9c08eec565
Clear memory cache when switching model and mode ( #1194 )
20230315.605
2023-03-15 22:18:26 -07:00
m68k-fr
2d2c523ac5
[Web] Upgrade Gradio to v3.21.0 ( #1188 )
...
Co-authored-by: powderluv <powderluv@users.noreply.github.com >
20230315.604
20230315.603
20230315.602
2023-03-15 10:14:49 -07:00
Abhishek Varma
f17b3128c0
[SD] Add LoRA inference to SD pipeline ( #1189 )
...
-- This commit adds LoRA inference to SD pipeline.
-- It also modifies txt2img to incorporate the new feature.
img2img, inpaint, outpaint, etc using Unet can also be extended in a
similar way.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
Co-authored-by: Abhishek Varma <abhishek@nod-labs.com >
2023-03-15 10:13:45 -07:00
Abhishek Varma
7c7e630099
[SD] Add fix for using latest diffusers + add scribble variant to Stencil ( #1191 )
...
* [SD] Add Scribble variant in Stencil
-- This commit adds scribble variant in Stencil.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
* [SD] Use latest diffusers
-- This commit points back to the latest diffusers and updates the
processing script to tackle the Pix2Pix import issue.
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
---------
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
Co-authored-by: Abhishek Varma <abhishek@nod-labs.com >
2023-03-15 10:13:20 -07:00
m68k-fr
2dd1491ec1
[Web] Add clear queue button ( #1192 )
2023-03-15 10:12:59 -07:00
Daniel Garvey
236357fb61
add missing import for shark_sd.spec ( #1190 )
...
L
2023-03-15 09:23:29 -05:00
Phoenix Meadowlark
7bc38719de
Add benchmark artifacts to .gitignore ( #1186 )
2023-03-14 15:19:06 -07:00
Daniel Garvey
bdbe992769
Add IREE_SAVE_TEMPS for import_debug command ( #1184 )
...
based on hf_model_id. Works on windows
2023-03-14 11:40:23 -07:00
Abhishek Varma
e6b925e012
[SD] Add Openpose to Stencil + image size issue fix ( #1181 )
...
-- This commit adds openpose model variant to stencil.
-- Fixes image size issue.
-- Also includes fix for the .exe bug introduced by https://github.com/nod-ai/SHARK/pull/1175
Signed-off-by: Abhishek Varma <abhishek@nod-labs.com >
Co-authored-by: Abhishek Varma <abhishek@nod-labs.com >
2023-03-14 10:30:52 -07:00
cstueckrath
771120b76c
workaround Gradio issue ( #1183 )
...
https://discord.com/channels/973663919757492264/975522729564446740/1085109774758191164
20230314.599
2023-03-14 01:27:24 -07:00
Boian Petkantchin
a8ce7680db
Add flag to augment the device allocator ( #1182 )
...
Example:
$ python my_app.py --device_allocator caching debug
This will wrap the device allocator with first caching allocator then
debug allocator.
$ python my_app.py --device_allocator caching
Only wrap with caching allocator.
Co-authored-by: Boian Petkantchin <boian@nod-labs.com >
20230313.598
2023-03-13 15:49:26 -07:00
Phaneesh Barwaria
b6dcf2401b
Stencil perf improvement ( #1179 )
...
* remove conditioning strength multiplier
* mod diffusers lib to v0.14.0
20230313.597
2023-03-13 14:37:38 -07:00
Daniel Garvey
62b5a9fd49
generate sharktank for apps dir ( #966 )
...
* merge confix resolution
* add support to other scripts
---------
Co-authored-by: dan <dan@nod-labs.com >
2023-03-13 10:54:15 -07:00
m68k-fr
2f133e9d5c
Fix png metadata ( #1178 )
2023-03-12 22:43:39 -07:00
powderluv
f898a1d332
Update README.md
2023-03-12 16:54:42 -07:00