update container

modelscope · Apr 24, 2024 · c89a92a · c89a92a
1 parent 06141ca
commit c89a92a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/docker/docker-compose.yml b/.github/workflows/docker/docker-compose.yml
@@ -1,7 +1,7 @@
 version: '3'
 services:
   ray-head:
-    image: data-juicer-unittest:v2
+    image: data-juicer-unittest:v0.2.0
     pull_policy: never
     command: ray start --head --dashboard-host 0.0.0.0 --include-dashboard true --block
     environment:
@@ -16,7 +16,7 @@ services:
     ports:
       - "6379:6379"
       - "8265:8265"
-    shm_size: "32G"
+    shm_size: "64G"
     deploy:
       resources:
         reservations:
@@ -26,7 +26,7 @@ services:
             capabilities: [gpu]
 
   ray-worker:
-    image: data-juicer-unittest:v2
+    image: data-juicer-unittest:v0.2.0
     pull_policy: never
     command: ray start --address=ray-head:6379 --block
     environment:
@@ -39,7 +39,7 @@ services:
       - ray-head
     networks:
       - ray-network
-    shm_size: "32G"
+    shm_size: "64G"
     deploy:
       resources:
         reservations:

diff --git a/.github/workflows/unittest-gpu.yml b/.github/workflows/unittest-gpu.yml
@@ -28,7 +28,7 @@ jobs:
     - name: Install data-juicer
       working-directory: .github/workflows/docker
       run: |
-        docker compose exec ray-head pip install -v -e .\[all\]
+        docker compose exec ray-head pip install -e .\[all\]
 
     - name: Remove docker compose
       working-directory: .github/workflows/docker