github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity
732 Commits 66 Branches 705 Tags
20221218.394
Commit Graph

64 Commits

Author SHA1 Message Date
Quinn Dawkins
2bc6de650d [SD] Add support for a compiled version of the discrete Euler scheduler (#657) 
* Add Shark version of euler scheduler

* Add Shark version of euler scheduler to web ui
 2022-12-17 19:25:43 -08:00
Gaurav Shukla
10160a066a [SD][WEB] Add vae tuned model in the SD web (#653) 
1. Add tuned vae model in the SD web.
2. Use tuned models in case of rdna3 cards.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-16 15:29:48 -08:00
Anush Elangovan
72976a2ece Import env vars first 2022-12-16 15:12:28 -08:00
Gaurav Shukla
72648aa9f2 Revert "[SD][WEB] Deduce vulkan-target-triple in the presence of multiple cards" 
This reverts commit 35e623deaf.
 2022-12-17 04:28:18 +05:30
Gaurav Shukla
35e623deaf [SD][WEB] Deduce vulkan-target-triple in the presence of multiple cards 
1. Get the correct vulkan-target-triple for a specified device in the
   presence of multiple cards.
2. Use tuned unet model for rdna3 cards.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-17 03:04:47 +05:30
Anush Elangovan
535d012ded Fix lint 2022-12-16 13:24:51 -08:00
Anush Elangovan
30fdc99f37 Set to enable llpc 
Use an env var to enable llpc
 2022-12-16 12:57:30 -08:00
Gaurav Shukla
bba06d0142 [SD][WEB] Avoid passing args to utils APIs 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-17 01:41:33 +05:30
Gaurav Shukla
e7e763551a [WEB][SD] Make unet tuned model default for rdna3 devices (#642) 2022-12-15 12:02:03 -08:00
Gaurav Shukla
a15cb140ae [WEB] Display the 512x512 image size 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-14 22:43:03 +05:30
Gaurav Shukla
e67ea31ee2 [SHARK][SD] Add --local_tank_cache flag in the stable diffusion 
This flag can be used to set local shark_tank cache directory.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-14 20:00:25 +05:30
Gaurav Shukla
986c126a5c [SHARK][SD] Add support for negative prompts 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-14 18:20:09 +05:30
Gaurav Shukla
0eee7616b9 [WEB] Launch only one SD version at a time 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-14 17:30:24 +05:30
Gaurav Shukla
d913453e57 [WEB] Update models to 8dec and also default values (#620) 
1. Update the models to 8 dec.
2. precision is default to `fp16` in CLI.
3. version is default to `v2.1base` in CLI as well as web.
4. The default scheduler is set to `EulerDiscrete` now.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-13 13:08:33 -08:00
Gaurav Shukla
5f5e0766dd [WEB] Add SD2.1 web support 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-13 21:36:01 +05:30
Gaurav Shukla
b62ee3fcb9 [WEB] Add schedulers in the web UI (#594) 
1. Add schedulers option in web UI.
2. Remove random seed checkbox as the same functionality can be achieved
   by passing -1(or any negative number) to the seed.

Signed-Off-by: Gaurav Shukla

Signed-off-by: Gaurav Shukla
 2022-12-08 13:53:20 -08:00
Ean Garvey
0225292a44 Remove print statements from compile utils (#593) 2022-12-08 13:40:47 -08:00
Quinn Dawkins
b3a42cd0b1 Don't do nchw-to-nhwc transpose for stable diffusion models (#592) 2022-12-08 12:19:23 -05:00
Gaurav Shukla
e3e1ca7cc6 [WEB] Fix seed when out of uint32 range 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-08 22:46:33 +05:30
Gaurav Shukla
57e417d174 [WEB] Fix web performance 
Set the iree flags before compilation.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-08 19:57:20 +05:30
Quinn Dawkins
dab9403b8f Fix slow conversion to image in SD web gui (#586) 2022-12-08 00:35:51 -05:00
Gaurav Shukla
e9e138c757 [WEB] Add random seed checkbox 
When True, it will not use user specified seed, instead will generate a
random seed.

Signed-Off-by: Gaurav Shukla
 2022-12-06 21:44:22 +05:30
Gaurav Shukla
29cc478525 [WEB] Add command line args to shark web 
1. Now the server can be launched with command line args.
2. The `precision` and `scheduler` parameters are now part of command
   line args instead of UI.
3. Add vae encode model wrapper.

Signed-Off-by: Gaurav Shukla
 2022-12-06 17:21:05 +05:30
Anush Elangovan
27b34f3929 Add gcs instead of gsutil 
Test .exe on AMD hardware.
 2022-12-05 22:17:58 -08:00
Daniel Garvey
8859853b41 Revert "Revert "find gsutil on linux (#557)" (#560)" (#561) 
This reverts commit 3c46021102.
 2022-12-05 20:27:43 -08:00
Daniel Garvey
3c46021102 Revert "find gsutil on linux (#557)" (#560) 
This reverts commit bba8646669.
 2022-12-05 21:53:47 -06:00
Daniel Garvey
bba8646669 find gsutil on linux (#557) 
* find gsutil on linux

* cleaned up downloader and ditched gsutil

Co-authored-by: dan <dan@nod-labs.com>
 2022-12-05 19:03:48 -08:00
Anush Elangovan
56109fe09b Add one click installer 
Build with pyinstaller web\shark_sd.spec
 2022-12-02 14:07:10 -08:00
Ean Garvey
ebb60019aa Minor formatting fix. (#538) 2022-12-03 00:17:31 +05:30
Anush Elangovan
8c158f2452 Fix onedir pyinstall 
Use relative paths for install

pyinstaller web/shark_sd.spec creates an exe
 2022-12-02 07:28:22 -08:00
Gaurav Shukla
a27ee141b3 [WEB] Fix few warnings and generate seed faster 
1. Fix gsutil warnings while copying multiple files.
2. Enhance random seed generation speed.
3. Add support for multiple schedulers.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-02 17:16:19 +05:30
Anush Elangovan
1106456651 Update cuda 11.7 nightly URL and add index.spec 2022-12-01 22:49:23 -08:00
Gaurav Shukla
a9bac0287d [WEB] Update to latest models. 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-12-01 22:55:31 +05:30
Gaurav Shukla
efbd3dc778 [WEB] Fix debug option and add random seed generation 
Signed-Off-by: Gaurav Shukla
 2022-12-01 21:08:34 +05:30
Gaurav Shukla
a5445866b8 [WEB] Update the iree flag 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-30 18:56:48 +05:30
Gaurav Shukla
3903b6ff0c [WEB] Enable Debug and disable live preview 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-29 22:39:53 +05:30
Gaurav Shukla
645b658da0 [WEB] Update model wrappers and scheduler 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-29 21:22:33 +05:30
Gaurav Shukla
3eb7965ca0 [WEB] Pressing Enter at prompt triggers Image generation 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-28 20:56:20 +05:30
Gaurav Shukla
88937fcb2f [WEB] Add vulkan-heap-block-size flag 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-24 16:58:27 +05:30
Gaurav Shukla
f4821d0d39 [WEB] Update seed calculation and model versions. 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-23 19:21:48 +05:30
Gaurav Shukla
275c032264 [WEB] Fix set_param prototype 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-22 20:15:45 +05:30
Gaurav Shukla
d88979fe19 [WEB] Enable guidance scale and update seed calculation 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-22 19:32:21 +05:30
Gaurav Shukla
d624940e12 Remove unnecessary torch_mlir import 
Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-21 21:47:21 +05:30
Gaurav Shukla
7763403b0e [WEB] Cache text-encoder and reorganize the codebase 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-21 17:21:12 +05:30
Gaurav Shukla
29a317dbb6 [WEB] Update SD styling and prompt loading. (#479) 
* [WEB] CSS changes to the web-ui (#465)

This commit updates UI with styling.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>

* [WEB] Update the title (#466)

* [WEB] Add support for long prompts (#467)

* [WEB] fix background color

Signed-Off-by: Gaurav Shukla

* [WEB] Remove long prompts support

It removes support to long prompts due to higher lag in loading long prompts.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs>

* [WEB] Update nod logo and enable debug feature.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
Signed-off-by: Gaurav Shukla
Signed-off-by: Gaurav Shukla <gaurav@nod-labs>
 2022-11-10 10:55:22 -08:00
Gaurav Shukla
88f8718635 [WEB] Load prompts from json 
The prompt examples will now be loaded from a json file `prompts.json`.

Signed-Off-by: Gaurav Shukla
 2022-11-02 20:52:34 +05:30
Gaurav Shukla
06ccfb0533 [WEB] Load vae and unet during server start up 
The vae and unet models(both fp16 and fp32 variant) can be loaded at
server startup in order to reduce web response time.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-01 23:11:52 +05:30
Gaurav Shukla
b18d75e3f7 [WEB] Use tuned version of UNET fp16 
This commit updates SD script in order to use the tuned version of Unet
fp16 model.

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-11-01 19:00:21 +05:30
Gaurav Shukla
a3fdfc81db [WEB] Minor changes in the shark web (#454) 
1. Default steps = 50.
2. Live preview will yield intermediate image at every 5 steps.
3. Add logs to .gitignore

Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-10-31 14:29:00 -07:00
Gaurav Shukla
f4c91df1df [WEB] Add pillow dependency (#453) 
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>

Signed-off-by: Gaurav Shukla <gaurav@nod-labs.com>
 2022-10-31 12:57:21 -07:00