From d606d4aa5292b7de9502304432f0ea7bf0dbfa38 Mon Sep 17 00:00:00 2001 From: Eugene Cheah Date: Sat, 3 Feb 2024 11:19:31 -0800 Subject: [PATCH] fixing docker ref --- docker/github-worker-cuda-12-1/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/github-worker-cuda-12-1/Dockerfile b/docker/github-worker-cuda-12-1/Dockerfile index 83b4942f..32871a53 100644 --- a/docker/github-worker-cuda-12-1/Dockerfile +++ b/docker/github-worker-cuda-12-1/Dockerfile @@ -1,5 +1,5 @@ # Temporary, until the rwkv package is public -FROM ghcr.io/RWKV/RWKV-infctx-trainer:env-cuda-12-1 +FROM ghcr.io/rwkv/rwkv-infctx-trainer:env-cuda-12-1 # FROM ghcr.io/rwkv/rwkv-infctx-trainer:env-cuda-12-1 # Install the github runner @@ -38,4 +38,4 @@ ENV RUNNER_REPO_URL="https://github.com/RWKV" # to account for the fact that we are running two runners # # This is only useful for high GPU, and high ram count machines -ENV RUNNER_LANE2="false" +ENV RUNNER_LANE2="false"