From d18edc8265d757a7fc0c23adf57db5dc6e00fcbd Mon Sep 17 00:00:00 2001
From: Maria Mirkina <maria@krai.ai>
Date: Thu, 11 Jan 2024 16:16:47 +0000
Subject: [PATCH] Added recipe for stable_diffusion model

---
 data_axs.json                                 |  3 +-
 .../data_axs.json                             | 29 +++++++++++++++++++
 2 files changed, 31 insertions(+), 1 deletion(-)
 create mode 100644 model_pytorch_stable_diffusion_recipe/data_axs.json

diff --git a/data_axs.json b/data_axs.json
index 4345f14..7dfda15 100644
--- a/data_axs.json
+++ b/data_axs.json
@@ -35,7 +35,8 @@
         "model_pytorch_gptj_cnndm_recipe": "model_pytorch_gptj_cnndm_recipe",
         "base_power_measurement_experiment": "base_power_measurement_experiment",
         "text_to_image_reference_using_torch_loadgen": "text_to_image_reference_using_torch_loadgen",
-        "cnndm_preprocessor": "cnndm_preprocessor"
+        "cnndm_preprocessor": "cnndm_preprocessor",
+        "model_pytorch_stable_diffusion_recipe": "model_pytorch_stable_diffusion_recipe"
     },
     "repo_name": "axs2mlperf",
     "submodules": false
diff --git a/model_pytorch_stable_diffusion_recipe/data_axs.json b/model_pytorch_stable_diffusion_recipe/data_axs.json
new file mode 100644
index 0000000..2f09c43
--- /dev/null
+++ b/model_pytorch_stable_diffusion_recipe/data_axs.json
@@ -0,0 +1,29 @@
+{
+    "_producer_rules": [
+        [ [ "downloaded", "pytorch_model", "model_name=stable_diffusion", "dtype=fp32" ], [["get_kernel"],["byname","downloader"],["download"]], {
+            "file_path": "stable_diffusion_fp32.zip",
+            "url": "https://cloud.mlcommons.org/index.php/s/DjnCSGyNBkWA4Ro/download"
+        } ],
+        [ [ "extracted",  "pytorch_model", "model_name=stable_diffusion", "dtype=fp32" ], [["get_kernel"],["byname","extractor"],["extract"]], {
+            "archive_format": "zip",
+            "archive_path": [ "^", "execute", [[
+                [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=stable_diffusion,dtype=fp32" ]] ],
+                [ "get_path" ]
+            ]] ],
+            "file_name": [ "extracted", "stable_diffusion_fp32" ]
+        } ],
+
+        [ [ "downloaded", "pytorch_model", "model_name=stable_diffusion", "dtype=fp16" ], [["get_kernel"],["byname","downloader"],["download"]], {
+            "file_path": "stable_diffusion_fp16.zip",
+            "url": "https://cloud.mlcommons.org/index.php/s/LCdW5RM6wgGWbxC/download"
+        } ],
+        [ [ "extracted",  "pytorch_model", "model_name=stable_diffusion", "dtype=fp16" ], [["get_kernel"],["byname","extractor"],["extract"]], {
+            "archive_format": "zip",
+            "archive_path": [ "^", "execute", [[
+                [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=stable_diffusion,dtype=fp16" ]] ],
+                [ "get_path" ]
+            ]] ],
+            "file_name": [ "extracted", "stable_diffusion_fp16" ]
+        } ]
+    ]
+}