diff --git a/docker-cuda/Dockerfile_base b/docker-cuda/Dockerfile_base
index c026f9df..8f7e2f63 100644
--- a/docker-cuda/Dockerfile_base
+++ b/docker-cuda/Dockerfile_base
@@ -5,6 +5,6 @@
 FROM mambaorg/micromamba
 WORKDIR /content/
 COPY environments /content/environments
-RUN micromamba install -y -n base -f /content/environments/huggingface.yml
+RUN micromamba install -y -n base -f /content/environments/base.yml
 USER root
 RUN apt update && apt install xorg -y
diff --git a/docker-cuda/Dockerfile_base_finetune b/docker-cuda/Dockerfile_base_finetune
index 4fd7d0ff..0b67a7db 100644
--- a/docker-cuda/Dockerfile_base_finetune
+++ b/docker-cuda/Dockerfile_base_finetune
@@ -1,9 +1,8 @@
 # This dockerfile is meant to serve as a rocm base image.  It registers the debian rocm package repository, and
 # installs the rocm-dev package.
 
-FROM mambaorg/micromamba
+FROM ebolam/koboldai_base
 WORKDIR /content/
-COPY environments /content/environments
-RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml
-USER root
-RUN apt update && apt install xorg -y
+RUN apt-get update
+RUN apt-get install git -y
+RUN /opt/conda/bin/pip install tensorflow-gpu git+https://github.com/finetuneanon/transformers.git
diff --git a/docker-cuda/Dockerfile_base_huggingface b/docker-cuda/Dockerfile_base_huggingface
new file mode 100644
index 00000000..47406cc1
--- /dev/null
+++ b/docker-cuda/Dockerfile_base_huggingface
@@ -0,0 +1,7 @@
+# This dockerfile is meant to serve as a rocm base image.  It registers the debian rocm package repository, and
+# installs the rocm-dev package.
+
+
+FROM ebolam/koboldai_base
+WORKDIR /content/
+RUN /opt/conda/bin/pip install transformers