Update Dockerfile

Dockerfile

@@ -14,6 +14,17 @@ WORKDIR /content
 USER user
 
 RUN pip3 install --upgrade pip
+
+# Add Hugging Face cache setup
+ENV HF_HOME=/content/huggingface
+ENV TRANSFORMERS_CACHE=$HF_HOME
+RUN mkdir -p $HF_HOME
+
+# Install transformers and download tokenizer
+RUN pip3 install --upgrade pip
+RUN pip install transformers==4.26.1
+RUN python -c "from transformers import CLIPTokenizer; CLIPTokenizer.from_pretrained('openai/clip-vit-large-patch14')"
+
 RUN pip install torchmetrics==0.11.4
 RUN pip install https://github.com/camenduru/stable-diffusion-webui-colab/releases/download/0.0.16/xformers-0.0.16+814314d.d20230119.A10G-cp310-cp310-linux_x86_64.whl
 RUN pip install --pre triton