From 34b3616eaa8aa777594c8517b085c96ffa5eb94f Mon Sep 17 00:00:00 2001 From: Eugene Cheah Date: Sat, 3 Feb 2024 11:08:02 -0800 Subject: [PATCH] base model tweak --- docker/github-worker-cuda-12-1/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/github-worker-cuda-12-1/Dockerfile b/docker/github-worker-cuda-12-1/Dockerfile index 93f5ecf3..2c623eb9 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-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