UPDATE

Dockerfile

@@ -23,15 +23,15 @@ RUN if [ "$MODELSCOPE_ENVIRONMENT" = "studio" ]; then \
     pnpm run build --mode target-rwkv-hf-space; \
   fi
 
-FROM nvidia/cuda:12.8.1-devel-ubuntu24.04 AS Backend
+FROM nvidia/cuda:12.4.0-devel-ubuntu22.04 AS Backend
 
 RUN <<EOF
 apt update
 apt install --no-install-recommends -y \
     build-essential \
     git \
-    cuda-nvcc-12-8 \
-    cuda-cudart-dev-12-8 \
+    cuda-nvcc-12-4 \
+    cuda-cudart-dev-12-4 \
     python3-dev \
     python3-pip \
     libpython3.10-dev