Skip to content

Commit

Permalink
Added weight_data_types and file_name for model_pytorch_stable_diffus…
Browse files Browse the repository at this point in the history
…ion_recipe downloadin by rclone
  • Loading branch information
maria-18-git committed Mar 21, 2024
1 parent 241ec15 commit f40c934
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions model_pytorch_stable_diffusion_recipe/data_axs.json
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,9 @@
"_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": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp32"
"url": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp32",
"weight_data_types": "fp32",
"file_name": [ "extracted", "stable_diffusion_fp32" ]
}, [ "input_data_types", "retraining", "weight_transformations" ] ],
[ [ "downloaded", "pytorch_model", "model_name=stable_diffusion", "dtype=fp32", "source?=via_wget" ], [["get_kernel"],["byname","downloader"],["download"]], {
"file_path": "stable_diffusion_fp32.zip",
Expand All @@ -22,7 +24,9 @@
[ [ "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": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp16"
"url": "mlc-inference:mlcommons-inference-wg-public/stable_diffusion_fp16",
"weight_data_types": "fp16",
"file_name": [ "extracted", "stable_diffusion_fp16" ]
}, [ "input_data_types", "retraining", "weight_transformations" ] ],
[ [ "downloaded", "pytorch_model", "model_name=stable_diffusion", "dtype=fp16", "source?=via_wget" ], [["get_kernel"],["byname","downloader"],["download"]], {
"file_path": "stable_diffusion_fp16.zip",
Expand Down

0 comments on commit f40c934

Please sign in to comment.