Daniel Garvey a4634632ba add generate_sharktank for stable_diffusion model defaults (#742) ...

Co-authored-by: dan <dan@nod-labs.com>
Co-authored-by: powderluv <powderluv@users.noreply.github.com>

2023-01-31 09:44:54 -08:00

..

ESRGAN

[SHARK_INFERENCE] Add ESRGAN model test file

2022-11-10 17:12:42 +05:30

stable_diffusion

add generate_sharktank for stable_diffusion model defaults (#742)

2023-01-31 09:44:54 -08:00

upscaler

This PR adds --import-mlir for f16 tensors without cuda.

2023-01-13 22:19:53 +05:30

albert_maskfill_pt.py

Level Zero Backend (#280)

2022-08-17 19:19:27 -07:00

albert_maskfill_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

bloom_tank.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

CLIPModel_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

gpt2_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

mhlo_example.py

Add shark_importer for torch_models. (#183)

2022-07-12 20:38:19 -07:00

minilm_benchmark_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

minilm_benchmark.py

Divide iree_utils and do module imports on function calls.

2022-06-22 14:17:33 +05:30

minilm_jit.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

minilm_tf_gpu_config.json

Add the option to use tuned model in shark_runner (#79)

2022-06-02 18:25:33 -07:00

minilm_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

resnest.py

Update unet_model to run on shark.

2022-08-17 13:16:02 +05:30

resnet50_fp16.py

Add FP16 Resnet50 script

2022-10-06 21:56:43 +05:30

resnet50_script.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

simple_dlrm.py

Add the dlrm_model in shark example. (#301)

2022-08-31 12:54:21 -07:00

sparse_arch.py

Add sparse architecture and test with torchrec SparseArch.

2022-09-09 21:49:30 +05:30

stable_diff_f16.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

stable_diff_tf.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00

stable_diff.py

Add option for extra arguments through SharkInference.compile (#408)

2022-10-19 15:32:48 -05:00

t5_tf.py

Enable XLA compiler for TF models (#484)

2022-11-13 20:10:47 -08:00

torch_vision_models_script.py

Divide iree_utils and do module imports on function calls.

2022-06-22 14:17:33 +05:30

unet_script.py

Update unet_model to run on shark.

2022-08-17 13:16:02 +05:30

v_diffusion.py

Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)

2022-12-05 20:27:43 -08:00