diff --git a/Dockerfile b/Dockerfile
index f6ebb238c..34b66a9ad 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM rust:latest as builder
+FROM rust:latest AS builder
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
@@ -9,7 +9,7 @@ COPY . .
 
 RUN cargo build --release --workspace --exclude mistralrs-pyo3
 
-FROM debian:bookworm-slim as base
+FROM debian:bookworm-slim AS base
 
 ENV HUGGINGFACE_HUB_CACHE=/data \
     PORT=80 \
diff --git a/Dockerfile.cuda-all b/Dockerfile.cuda-all
index ee2da7354..90a20be5a 100644
--- a/Dockerfile.cuda-all
+++ b/Dockerfile.cuda-all
@@ -21,7 +21,7 @@ ARG FEATURES="cuda cudnn"
 ENV RAYON_NUM_THREADS=4
 RUN RUSTFLAGS="-Z threads=4" cargo build --release --workspace --exclude mistralrs-pyo3 --features "${FEATURES}"
 
-FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 as base
+FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 AS base
 
 ENV HUGGINGFACE_HUB_CACHE=/data \
     PORT=80 \