base model tweak

RWKV · Feb 3, 2024 · 34b3616 · 34b3616
1 parent aca143b
commit 34b3616
Showing 1 changed file with 1 addition and 1 deletion.
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-lm-lora: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