Prashant Kumar
3c4c4231ce
Add new args.
2022-11-29 18:18:04 +05:30
Prashant Kumar
d0eef19eba
Remove the lms versions as they were redundant.
...
Tested with the DPM scheduler.
2022-11-29 15:05:05 +05:30
Prashant Kumar
74aeb55733
Add support for different schedulers.
...
Initial support for adding schedulers. This verifies the model running
with the PNDM scheduler too.
2022-11-28 22:12:09 +05:30
Quinn Dawkins
32a2ec432d
[Stable Diffusion] Revive the tuned model ( #506 )
2022-11-23 15:42:24 -05:00
Prashant Kumar
fdf2aa54ef
Update the sd models.
2022-11-22 23:09:04 +05:30
Phaneesh Barwaria
e67bcffea7
add vulkan-heap-block-size flag ( #498 )
2022-11-22 13:30:25 +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
Prashant Kumar
88c58244b9
Update stable diffusion models to point to new location.
2022-11-18 19:39:21 +05:30
Prashant Kumar
7b1f04d121
Changes incorporating the recent torch_mlir compile api changes.
2022-11-15 15:25:37 +05:30
Ean Garvey
45677c1e23
Install torch version required by torch-mlir when setting up importer venv. ( #486 )
2022-11-14 14:01:01 +05:30
Phaneesh Barwaria
d9f4a9954a
modify to get correct target triple ( #485 )
2022-11-13 20:13:44 -08:00
mariecwhite
ec461a4456
Enable XLA compiler for TF models ( #484 )
2022-11-13 20:10:47 -08:00
Phaneesh Barwaria
749a2c2dec
add support for choosing vulkan device ( #439 )
2022-11-12 14:00:41 -08:00
Abhishek Varma
2f36de319a
[SHARK_INFERENCE] Add ESRGAN model test file
...
-- This commit adds ESRGAN model test file to SHARK_INFERENCE.
Signed-off-by: Abhishek Varma <abhishek@nod-ai.com >
2022-11-10 17:12:42 +05:30
Quinn Dawkins
2005bce419
Fix flags for untuned Stable Diffusion FP16 model ( #478 )
2022-11-09 21:31:10 -05:00
Prashant Kumar
1cdf301c14
Update the guidance parameter argument and add the int8 version of the
...
stable diffusion model.
2022-11-08 23:14:44 +05:30
yzhang93
9a86e5c476
Fix dispatch benchmarking tool ( #460 )
2022-11-08 09:37:12 -08:00
Prashant Kumar
18689afc1a
Make separate function for each model.
2022-11-07 20:20:38 +05:30
Prashant Kumar
a081733a42
Add the clip text shark_model. ( #458 )
2022-11-02 00:08:33 -07:00
Quinn Dawkins
3e7efaa048
Switch stable diffusion to the new tuned model ( #455 )
2022-10-31 15:15:31 -07:00
Prashant Kumar
32e1ba8c0d
Adding batch_size support for stable diffusion.
2022-11-01 00:57:52 +05:30
Quinn Dawkins
239c19eb12
Update Stable diffusion script to enable use of tuned models ( #443 )
2022-10-29 01:42:49 -04:00
Eliasj42
7f37599a60
Added a dispatch benchmarking tool ( #441 )
...
To produce benchmarks of individual dispatches, you can add --dispatch_benchmarks=All --dispatch_benchmarks_dir=<output_dir> to your command line argument.
Co-authored-by: Elias Joseph <elias@nod-labs.com >
2022-10-28 14:31:03 -07:00
Prashant Kumar
77c9a2c5ea
Add profiling vulkan_device info and minor changes to reflect upstream
...
changes.
2022-10-28 18:02:07 +05:30
Stanley Winata
01fdf5ee16
[example][SD] compile fp16 with iree-spirv-unify-aliased-resources ( #436 )
2022-10-27 05:12:28 -07:00
Quinn Dawkins
fbd77dc936
Enable iterator space fusion for SD ( #432 )
2022-10-26 01:08:26 -04:00
Quinn Dawkins
cdc6dd19e3
Force stable diffusion fp16 and fp32 to generate images with similar noise ( #431 )
2022-10-25 17:28:18 -04:00
PhaneeshB
fd578a48a9
add cli args for vulkan target triple
2022-10-25 21:47:26 +05:30
Quinn Dawkins
1d33913d48
Add option to save and load precompiled flatbuffer ( #425 )
2022-10-23 16:24:09 -07:00
Prashant Kumar
a48eaaed20
Pass the flags to vae.
2022-10-23 23:57:48 +05:30
Prashant Kumar
2741b8be53
Pass the flags to vae. ( #422 )
2022-10-23 11:23:13 -07:00
Anush Elangovan
4f906a265c
Fix lint
2022-10-22 12:43:52 -07:00
Anush Elangovan
0dff8d7af0
Simple download script to prime the hf model cache
2022-10-21 17:42:05 -07:00
Quinn Dawkins
4f0d0d8167
Update vulkan gui README for iree-vulkan-gui + Stable Diffusion ( #399 )
2022-10-21 14:02:40 -04:00
Vivek Khandelwal
d513060b21
Add params for Stable Diffusion ( #420 )
2022-10-21 23:11:09 +05:30
Prashant Kumar
d1a25ce4f3
Update stable_args.py
2022-10-21 17:26:31 +05:30
Quinn Dawkins
b448770ec2
Add ms/iter timing for stable diffusion script ( #414 )
2022-10-20 13:32:37 -04:00
Prashant Kumar
5fe22a7980
Minor fix.
2022-10-20 22:57:22 +05:30
Prashant Kumar
38ae6b5af4
Add stable_diffusion fp16 and fp32 with args.
2022-10-20 21:47:11 +05:30
Quinn Dawkins
7be1d7d0be
Add option for extra arguments through SharkInference.compile ( #408 )
2022-10-19 15:32:48 -05:00
Prashant Kumar
0d74c873f0
Add stable_diff_f16 version. ( #407 )
2022-10-19 10:04:24 -07:00
Vivek Khandelwal
0e217d6180
Add Stable Diffusion Img2Img model script
2022-10-13 21:56:46 +05:30
Quinn Dawkins
d93d0783a8
Add script for tensorflow stable diffusion ( #391 )
2022-10-10 12:01:49 -04:00
Vivek Khandelwal
70a29b03e0
Add FP16 Resnet50 script
2022-10-06 21:56:43 +05:30
Quinn Dawkins
33b53e7caf
Add flag for specifying the vae mlir file location in stable diffusion ( #381 )
2022-09-30 00:37:58 -04:00
Prashant Kumar
c7b2d39ab2
Update stable_diff to contain vae.
2022-09-26 20:11:43 +05:30
Prashant Kumar
b9c8985047
Add sharkdynamo which combines shark with torchdynamo.
...
-- Adds graph breaks when necessary.
-- Even for loops are supported.
2022-09-23 22:40:02 +05:30
Ean Garvey
d82b305781
Fix issues with loading .vmfb into SharkInference
2022-09-23 09:53:13 +05:30
Prashant Kumar
991e7043d1
Add stable diffusion model.
2022-09-22 13:40:51 +05:30
Vivek Khandelwal
864723a473
add bloom model example
2022-09-15 18:23:09 +05:30