Merge pull request #33 from replicate/hf-push-in-test

Add HF upload to train test
replicate · Sep 16, 2024 · cd8dec9 · cd8dec9
2 parents acc81ad + 2e9eb2e
commit cd8dec9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/.github/workflows/push.yaml b/.github/workflows/push.yaml
@@ -36,6 +36,12 @@ jobs:
       run: |
         pip install git+https://github.com/replicate/cog-safe-push.git
 
+    - name: "Inject secrets into cog-safe-push.yaml"
+      env:
+        HF_TOKEN: ${{ secrets.HF_WRITE_TO_TEST_FINE_TUNER_INTEGRATION }}
+      run: |
+        envsubst < cog-safe-push.yaml.tpl > cog-safe-push.yaml
+
     - name: Run cog-safe-push to test the trainer and predictor and optionally push to production
       env:
         ANTHROPIC_API_KEY: ${{ secrets.ANTHROPIC_API_KEY }}

diff --git a/cog-safe-push.yaml → cog-safe-push.yaml.tpl b/cog-safe-push.yaml → cog-safe-push.yaml.tpl
@@ -112,8 +112,10 @@ train:
 
     # tiny training with autocaption
     - inputs:
-        input_images: "https://replicate.delivery/pbxt/LbXqvBzG3Bfo6CDkIkBNo8a8v6GWZGQrqxBrgeRpaQ1knkyD/me-tiny.zip"
+        input_images: https://replicate.delivery/pbxt/LbXqvBzG3Bfo6CDkIkBNo8a8v6GWZGQrqxBrgeRpaQ1knkyD/me-tiny.zip
         steps: 50
         autocaption: true
         trigger_word: NDRS
+        hf_repo_id: replicate/test-flux-fine-tuner-integration
+        hf_token: "$HF_TOKEN"
       match_prompt: "A dictionary of a version and weights which are a .tar file"