Added recipe for stable_diffusion model

krai · Jan 11, 2024 · d18edc8 · d18edc8
1 parent 8da7610
commit d18edc8
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git 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
@@ -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" ]
+        } ]
+    ]
+}