Added correct url for rclone downoading(SDXl, GPT-J)

krai · Feb 20, 2024 · 624b225 · 624b225
1 parent 2a3cdbb
commit 624b225
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/model_pytorch_gptj_cnndm_recipe/data_axs.json b/model_pytorch_gptj_cnndm_recipe/data_axs.json
@@ -3,7 +3,7 @@
         [ [ "extracted", "pytorch_model", "model_name=gptj_cnndm", "source?=via_rclone" ], [["get_kernel"],["byname","downloader"],["download"]], {
             "downloading_tool_query": "shell_tool,can_download_url_from_rclone",
             "file_path": [ "extracted", "gpt-j", "checkpoint-final" ],
-            "url": "gpt-j"
+            "url": "mlc-inference:mlcommons-inference-wg-public/gpt-j"
         } ],
 
         [ [ "downloaded", "pytorch_model", "model_name=gptj_cnndm", "source?=via_wget" ], [["get_kernel"],["byname","downloader"],["download"]], {
@@ -13,7 +13,7 @@
             "md5b": "470d09e35f9afa75f79e70d689d0b96a"
         } ],
 
-        [ [ "extracted",  "pytorch_model", "model_name=gptj_cnndm" ], [["get_kernel"],["byname","extractor"],["extract"]], {
+        [ [ "extracted",  "pytorch_model", "model_name=gptj_cnndm", "source?=via_wget" ], [["get_kernel"],["byname","extractor"],["extract"]], {
             "archive_format": "zip",
             "archive_path": [ "^", "execute", [[
                 [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=gptj_cnndm,source=via_wget" ]] ],

diff --git a/model_pytorch_stable_diffusion_recipe/data_axs.json b/model_pytorch_stable_diffusion_recipe/data_axs.json
@@ -2,7 +2,7 @@
     "_producer_rules": [
         [ [ "extracted", "pytorch_model", "model_name=stable_diffusion", "dtype=fp32", "source?=via_rclone" ], [["get_kernel"],["byname","downloader"],["download"]], {
             "downloading_tool_query": "shell_tool,can_download_url_from_rclone",
-            "url": "stable_diffusion_fp32"
+            "url": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp32"
         } ],
         [ [ "downloaded", "pytorch_model", "model_name=stable_diffusion", "dtype=fp32", "source?=via_wget" ], [["get_kernel"],["byname","downloader"],["download"]], {
             "file_path": "stable_diffusion_fp32.zip",
@@ -11,7 +11,7 @@
         [ [ "extracted",  "pytorch_model", "model_name=stable_diffusion", "dtype=fp32", "source?=via_wget" ], [["get_kernel"],["byname","extractor"],["extract"]], {
             "archive_format": "zip",
             "archive_path": [ "^", "execute", [[
-                [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=stable_diffusion,dtype=fp32" ]] ],
+                [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=stable_diffusion,dtype=fp32,source=via_wget" ]] ],
                 [ "get_path" ]
             ]] ],
             "url": "https://cloud.mlcommons.org/index.php/s/DjnCSGyNBkWA4Ro/download",
@@ -22,7 +22,7 @@
         [ [ "extracted", "pytorch_model", "model_name=stable_diffusion", "dtype=fp16", "source?=via_rclone" ], [["get_kernel"],["byname","downloader"],["download"]], {
             "downloading_tool_query": "shell_tool,can_download_url_from_rclone",
             "file_path": "stable_diffusion_fp16",
-            "url": "stable_diffusion_fp16"
+            "url": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp16"
         } ],
         [ [ "downloaded", "pytorch_model", "model_name=stable_diffusion", "dtype=fp16", "source?=via_wget" ], [["get_kernel"],["byname","downloader"],["download"]], {
             "file_path": "stable_diffusion_fp16.zip",
@@ -31,7 +31,7 @@
         [ [ "extracted",  "pytorch_model", "model_name=stable_diffusion", "dtype=fp16", "source?=via_wget" ], [["get_kernel"],["byname","extractor"],["extract"]], {
             "archive_format": "zip",
             "archive_path": [ "^", "execute", [[
-                [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=stable_diffusion,dtype=fp16" ]] ],
+                [ "byquery", [[ "^^", "substitute", "downloaded,pytorch_model,model_name=stable_diffusion,dtype=fp16,source=via_wget" ]] ],
                 [ "get_path" ]
             ]] ],
             "url":  "https://cloud.mlcommons.org/index.php/s/LCdW5RM6wgGWbxC/download",