From 25456282d0580e1160423da0a91b058cd0903a2a Mon Sep 17 00:00:00 2001
From: Mishig <dmishig@gmail.com>
Date: Mon, 29 Jul 2024 12:17:45 +0200
Subject: [PATCH] Change doc-builder workflow container (#508)

* Change doc-builder workflow container

* trigger the action
---
 .github/workflows/build_documentation.yml    | 1 +
 .github/workflows/build_pr_documentation.yml | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/.github/workflows/build_documentation.yml b/.github/workflows/build_documentation.yml
index e69506fd..8b501942 100644
--- a/.github/workflows/build_documentation.yml
+++ b/.github/workflows/build_documentation.yml
@@ -18,6 +18,7 @@ jobs:
       package_path: safetensors/bindings/python/
       version_tag_suffix: bindings/python/py_src/
       install_rust: true
+      custom_container: huggingface/transformers-doc-builder
     secrets:
       token: ${{ secrets.HUGGINGFACE_PUSH }}
       hf_token: ${{ secrets.HF_DOC_BUILD_PUSH }}
diff --git a/.github/workflows/build_pr_documentation.yml b/.github/workflows/build_pr_documentation.yml
index d9f58812..efafcd48 100644
--- a/.github/workflows/build_pr_documentation.yml
+++ b/.github/workflows/build_pr_documentation.yml
@@ -5,6 +5,7 @@ on:
     paths:
       - "docs/**"
       - "bindings/python/py_src/**"
+      - ".github/workflows/build_pr_documentation.yml"
 
 concurrency:
   group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
@@ -20,3 +21,4 @@ jobs:
       package_path: safetensors/bindings/python/
       version_tag_suffix: bindings/python/py_src/
       install_rust: true
+      custom_container: huggingface/transformers-doc-builder