feat: Add GitHub Container Registry

dragonflyoss · Oct 11, 2023 · d5570ce · d5570ce
1 parent ec2c435
commit d5570ce
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
@@ -46,6 +46,13 @@ jobs:
           username: ${{ secrets.DOCKER_USERNAME }}
           password: ${{ secrets.DOCKER_PASSWORD }}
 
+      - name: Login to GitHub Container Registry
+        uses: docker/login-action@v3
+        with:
+          registry: ghcr.io
+          username: ${{ github.repository_owner }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
       - name: Push to Registry
         uses: docker/build-push-action@v5
         with:
@@ -60,5 +67,6 @@ jobs:
             VERSION=${{ steps.get_version.outputs.VERSION }}
           tags: |
             dragonflyoss/torchserve-dragonfly:${{ steps.get_version.outputs.VERSION }}
+            ghcr.io/${{ env.IMAGE_REPOSITORY }}/${{ matrix.module }}:${{ steps.get_version.outputs.VERSION }}
           push:
             true
diff --git a/src/main/java/org/pytorch/serve/plugins/dragonfly/Dragonfly.java b/src/main/java/org/pytorch/serve/plugins/dragonfly/Dragonfly.java
@@ -34,10 +34,6 @@
 
 /**
  * The Dragonfly endpoint for the Model Server.
- *
- * <p>param urlPattern The endpoint name. param endpointType The type of API, Management API port is
- * 8081. param description The function of endpoint.
- *
  * @see DragonflyModelRequest
  * @see ModelRegisterUtils
  * @throws RuntimeException If there's an error during download, model registration, or any other