Merge pull request #115 from Datura-ai/main

hotfix(validaor): update MAX_GPU_COUNT to 14
Datura-ai · Dec 5, 2024 · 17f6f61 · 17f6f61
2 parents bfe95c1 + 8bdec1c
commit 17f6f61
Show file tree

Hide file tree

Showing 5 changed files with 213 additions and 199 deletions.
diff --git a/neurons/validators/Dockerfile b/neurons/validators/Dockerfile
@@ -48,6 +48,6 @@ RUN echo "export PYTHONPATH=$PYTHONPATH" >> ~/.bash_profile
 COPY --from=base-image /root/app/ /root/app/
 COPY --from=base-image /opt/pypackages/ /opt/pypackages/
 
-LABEL version="3.2.5"
+LABEL version="3.2.6"
 
 CMD ["bash", "run.sh"]
diff --git a/neurons/validators/Dockerfile.runner b/neurons/validators/Dockerfile.runner
@@ -3,7 +3,7 @@ WORKDIR /root/validator
 COPY docker-compose.app.yml docker-compose.yml
 COPY entrypoint.sh /entrypoint.sh
 
-LABEL version="3.2.5"
+LABEL version="3.2.6"
 
 RUN chmod u+x /entrypoint.sh
 ENTRYPOINT ["/entrypoint.sh"]
diff --git a/neurons/validators/src/services/const.py b/neurons/validators/src/services/const.py
@@ -31,7 +31,7 @@
 UPLOAD_SPEED_WEIGHT = 0.05
 DOWNLOAD_SPEED_WEIGHT = 0.05
 
-MAX_GPU_COUNT = 8
+MAX_GPU_COUNT = 14
 
 UNRENTED_MULTIPLIER = 0.1
 

diff --git a/neurons/validators/src/services/docker_service.py b/neurons/validators/src/services/docker_service.py
@@ -331,9 +331,9 @@ async def delete_container(
             client_keys=[pkey],
             known_hosts=None,
         ) as ssh_client:
-            await ssh_client.run(f"docker stop {payload.container_name}")
+            # await ssh_client.run(f"docker stop {payload.container_name}")
             await ssh_client.run(f"docker rm {payload.container_name} -f")
-            await ssh_client.run(f"docker volume rm {payload.volume_name}")
+            await ssh_client.run(f"docker volume rm {payload.volume_name} -f")
 
             logger.info(
                 _m(