diff --git a/Dockerfile b/Dockerfile
index f3bb6d2..5dead65 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,12 +2,8 @@ FROM python:3.10
 
 WORKDIR /taleweave
 
-# COPY requirements/cpu.txt /taleweave/requirements/cpu.txt
-# RUN pip install --no-cache-dir -r requirements/cpu.txt
-
 COPY requirements/base.txt /taleweave/requirements/base.txt
 RUN pip install --no-cache-dir -r requirements/base.txt
-RUN pip install --no-cache-dir --index-url https://test.pypi.org/simple/ packit_llm==0.1.0
 
 COPY taleweave/ /taleweave/taleweave/
 
diff --git a/requirements/base.txt b/requirements/base.txt
index 661e258..7126524 100644
--- a/requirements/base.txt
+++ b/requirements/base.txt
@@ -8,6 +8,7 @@ langchain-core==0.1.50
 mistletoe==1.3.0
 numpy==1.26.4
 # packit-llm==0.1.0
+git+https://github.com/ssube/packit.git
 pillow==10.3.0
 pydantic==2.7.1
 pydantic_core==2.18.2