Spaces:

nroggendorff
/

tune-diffuser

Paused

nroggendorff commited on Nov 13

Commit

96e5830

verified ·

1 Parent(s): 8af79ef

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -1,5 +1,4 @@
-FROM python:latest
 ARG OUTPUT_DIR=zelda-style
 ARG INPUT_DS=nroggendorff/zelda-style
@@ -15,7 +14,7 @@ COPY . .
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir --prefer-binary \
-      diffusers[torch] \
       transformers \
       accelerate \
       safetensors \

+FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
 ARG OUTPUT_DIR=zelda-style
 ARG INPUT_DS=nroggendorff/zelda-style
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir --prefer-binary \
+      "git+https://github.com/huggingface/diffusers.git@main#egg=diffusers" \
       transformers \
       accelerate \
       safetensors \