Steveeeeeeen HF staff commited on
Commit
68a291c
·
verified ·
1 Parent(s): cde71ce

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -48,6 +48,12 @@ RUN git clone https://github.com/Zyphra/Zonos.git && cd Zonos
48
  # Copy application files
49
  WORKDIR /app/Zonos
50
 
 
 
 
 
 
 
51
  # Install Python dependencies
52
  RUN uv pip install --system -e . && uv pip install --system -e .[compile]
53
 
 
48
  # Copy application files
49
  WORKDIR /app/Zonos
50
 
51
+ # Set environment variable for Triton cache
52
+ ENV TRITON_CACHE_DIR=/tmp/.triton
53
+
54
+ # Ensure Triton cache directory is writable
55
+ RUN mkdir -p $TRITON_CACHE_DIR && chmod -R 777 $TRITON_CACHE_DIR
56
+
57
  # Install Python dependencies
58
  RUN uv pip install --system -e . && uv pip install --system -e .[compile]
59