fix: test and HxW

runpod-workers · Aug 8, 2023 · 01255df · 01255df
1 parent 859082f
commit 01255df
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/CI-test_worker.yml b/.github/workflows/CI-test_worker.yml
@@ -22,7 +22,7 @@ jobs:
         id: deploy
         uses: fjogeleit/http-request-action@v1
         with:
-          url: "https://api.runpod.ai/v2/${{ secrets.RUNPOD_ENDPOINT }}/run"
+          url: "https://api.runpod.ai/v2/${{ vars.RUNNER_24GB }}/run"
           method: "POST"
           customHeaders: '{"Content-Type": "application/json"}'
           bearerToken: ${{ secrets.RUNPOD_API_KEY }}
@@ -64,7 +64,7 @@ jobs:
       - name: Shutdown Worker
         uses: fjogeleit/http-request-action@v1
         with:
-          url: "https://api.runpod.ai/v2/${{ secrets.RUNPOD_ENDPOINT }}/cancel/${{ needs.initialize_worker.outputs.id }}"
+          url: "https://api.runpod.ai/v2/${{ vars.RUNNER_24GB }}/cancel/${{ needs.initialize_worker.outputs.id }}"
           method: "POST"
           customHeaders: '{"Content-Type": "application/json"}'
           bearerToken: ${{ secrets.RUNPOD_API_KEY }}
diff --git a/src/rp_schemas.py b/src/rp_schemas.py
@@ -11,12 +11,12 @@
     'height': {
         'type': int,
         'required': False,
-        'default': 512
+        'default': 1024
     },
     'width': {
         'type': int,
         'required': False,
-        'default': 512
+        'default': 1024
     },
     'num_inference_steps': {
         'type': int,