KarthikAI commited on
Commit
42a9c43
·
verified ·
1 Parent(s): 5ef6768

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -0
Dockerfile CHANGED
@@ -12,6 +12,13 @@ RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
12
  # Download models and adapters at build time
13
  # You can replace MODEL_ID or adapter repos as needed
14
 
 
 
 
 
 
 
 
15
  # -- Download Stable Diffusion v1-5 weights --
16
  # RUN python -c "from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype='float32')"
17
  RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype=torch.float32)"
 
12
  # Download models and adapters at build time
13
  # You can replace MODEL_ID or adapter repos as needed
14
 
15
+ ENV CACHE_DIR="/data"
16
+ # ENV TRANSFORMERS_CACHE="/data"
17
+ # ENV DEEPFACE_HOME="/data"
18
+ # ENV INSIGHTFACE_HOME="/data/.insightface"
19
+ RUN mkdir -p /data && \
20
+ chmod -R 777 /data
21
+
22
  # -- Download Stable Diffusion v1-5 weights --
23
  # RUN python -c "from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype='float32')"
24
  RUN python -c "import torch; from diffusers import StableDiffusionImg2ImgPipeline; StableDiffusionImg2ImgPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', cache_dir='./models', torch_dtype=torch.float32)"