fixing docker ref

RWKV · Feb 3, 2024 · d606d4a · d606d4a
1 parent 2e33f4b
commit d606d4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"