diff --git a/animation/hallo-83dd4eb/docker-56038fd/Dockerfile b/animation/hallo-83dd4eb/docker-56038fd/Dockerfile
index 670e6fe..2db3b16 100644
--- a/animation/hallo-83dd4eb/docker-56038fd/Dockerfile
+++ b/animation/hallo-83dd4eb/docker-56038fd/Dockerfile
@@ -29,7 +29,8 @@ RUN git clone https://github.com/fudan-generative-vision/hallo.git && \
 # Install the dependencies for hallo
 WORKDIR /hallo
 RUN source /venv/bin/activate && \
-    pip3 install -r requirements.txt && \
+    pip3 install -r requirements.txt \
+    --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ --extra-index-url https://download.pytorch.org/whl/cu121/ && \
     pip3 install . && \
     pip3 install huggingface_hub && \
     deactivate
diff --git a/animation/hallo-83dd4eb/docker-56038fd/docker-bake.hcl b/animation/hallo-83dd4eb/docker-56038fd/docker-bake.hcl
index 69f6607..3cd6bed 100644
--- a/animation/hallo-83dd4eb/docker-56038fd/docker-bake.hcl
+++ b/animation/hallo-83dd4eb/docker-56038fd/docker-bake.hcl
@@ -17,12 +17,9 @@ variable "CU_VERSION" {
     default = "121"
 }
 
-variable "BASE_IMAGE_REPOSITORY" {
-    default = "ashleykza/runpod-base"
-}
 
-variable "BASE_IMAGE_VERSION" {
-    default = "1.7.0"
+variable "BASE_IMAGE" {
+    default = "12.2.2-cudnn8-devel-ubuntu22.04"
 }
 
 variable "CUDA_VERSION" {
diff --git a/animation/hallo-83dd4eb/requirements.txt b/animation/hallo-83dd4eb/requirements.txt
index 5c4a9a5..d46936e 100644
--- a/animation/hallo-83dd4eb/requirements.txt
+++ b/animation/hallo-83dd4eb/requirements.txt
@@ -1,5 +1,3 @@
---find-links https://download.pytorch.org/whl/torch_stable.html
-
 accelerate==0.28.0
 audio-separator==0.17.2
 av==12.1.0