up

Dockerfile

@@ -38,6 +38,16 @@ ENV HOME=/home/user \
 
 RUN pip3 install --no-cache-dir --upgrade --pre -r /code/requirements.txt
 
+# Prepare the model directory
+
+RUN mkdir -p $HOME/app/lora_weights
+
+WORKDIR $HOME/app/lora_weights
+
+# Download some weights
+
+RUN curl -LJ https://drive.google.com/file/d/1_-kEVFw_LnV1J2Nho6nZt4PUbymamypK/view?usp=drive_link -o PixelArtRedmond15V-PixelArt-PIXARFK.safetensors
+
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app