first commit

Author: tazarov

.github/workflows/release.yaml

@@ -0,0 +1,42 @@
+name: Build Images
+
+on:
+  release:
+    types: [ created ]
+permissions:
+  actions: write
+  packages: write
+
+env:
+  IMAGE_NAME: "amikos/hf-model-downloader"
+
+jobs:
+  build-images:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+        with:
+          fetch-depth: 0
+      - name: Log in to the Container registry
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKERHUB_TOKEN }}
+          password: ${{ secrets.DOCKERHUB_USERNAME }}
+      - name: Docker meta
+        id: meta
+        uses: docker/metadata-action@v4
+        with:
+          images: ${{ env.IMAGE_NAME }}
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+      - name: Build and push release Docker image
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          file: Dockerfile
+          push: true
+          platforms: linux/amd64,linux/arm64
+          tags: "${{ env.IMAGE_NAME }}:${{ github.event.release.tag_name }},${{ env.IMAGE_NAME }}:latest"

.gitignore

@@ -0,0 +1,10 @@
+FROM python:alpine3.12
+
+LABEL maintainer="[email protected]"
+
+
+COPY docker-entrypoint.sh /
+RUN pip install 'huggingface_hub[cli]' && \
+    chmod +x /docker-entrypoint.sh
+
+ENTRYPOINT [ "/docker-entrypoint.sh" ]

Dockerfile

@@ -0,0 +1,22 @@
+# HuggingFace Hub Model Downloader
+
+The purpose of this docker image is to provide a convenient way to download models from the HuggingFace Hub using HF's CLI.
+
+## Usage
+
+```bash
+docker run -it --rm -v $(PWD):/models amikos/hf-model-downloader  sentence-transformers/all-MiniLM-L6-v2
+```
+
+With HF Token:
+
+```bash
+export HF_TOKEN="your_huggingface_token"
+docker run -it --rm -e HF_TOKEN="$HF_TOKEN" -v $(PWD):/models amikos/hf-model-downloader  sentence-transformers/all-MiniLM-L6-v2
+```
+
+## Building
+
+```bash
+docker build -t <tag> .
+```

README.md

@@ -0,0 +1,15 @@
+#!/bin/sh
+
+for model in "$@"
+do
+    # Sanitize the model name
+    sanitized_model=$(printf '%s' "$model" | sed 's/[^a-zA-Z0-9._-]/_/g')
+    token=""
+    # Add token if HF_TOKEN is defined
+    if [ -n "$HF_TOKEN" ]; then
+        token="--token ${HF_TOKEN}"
+        echo "Found HF token, using it to download $model"
+    fi
+    # Use printf to safely pass arguments
+    printf '%s\0' "--local-dir" "/models/$sanitized_model" "$model" | xargs -0 huggingface-cli download $token
+done