From c20bdb4b5f36d70abe4322d7feba58089a025856 Mon Sep 17 00:00:00 2001
From: Maria Mirkina <maria@krai.ai>
Date: Thu, 8 Feb 2024 04:03:11 -0600
Subject: [PATCH] Supporting of lfs and GIT_LFS_SKIP_SMUDGE in
 stable_diffusion_xl_git_recipe

---
 stable_diffusion_xl_git_recipe/data_axs.json | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/stable_diffusion_xl_git_recipe/data_axs.json b/stable_diffusion_xl_git_recipe/data_axs.json
index 4328bd7..22d7675 100644
--- a/stable_diffusion_xl_git_recipe/data_axs.json
+++ b/stable_diffusion_xl_git_recipe/data_axs.json
@@ -1,7 +1,19 @@
 {
     "_producer_rules": [
-        [ [ "git_repo", "repo_name=stable_diffusion_xl_base_1.0_git" ], [["get_kernel"],["byname","git"],["clone"]], {
-        }, [ "url" ] ]
+        [ [ "git_repo", "repo_name=stable_diffusion_xl_base_1.0_git" ], [
+                ["get_kernel"],
+                ["byname","git"],
+                ["clone"],
+                ["run",[],{
+                        "cmd_key": "generic",
+                        "rest_of_cmd": "lfs pull -I vae_decoder/model.onnx",
+                        "capture_output": false,
+                        "get_and_return_on_success": "__entry__"
+                    }]
+            ], {}, [ "url", "env" ] ]
     ],
-    "url": "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0"
+    "url": "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
+    "env": {
+        "GIT_LFS_SKIP_SMUDGE": "1"
+    }
 }