Spaces:
Paused
Paused
Create Dockerfile
Browse files- Dockerfile +15 -0
Dockerfile
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM python:latest
|
| 2 |
+
|
| 3 |
+
WORKDIR /app
|
| 4 |
+
|
| 5 |
+
COPY . .
|
| 6 |
+
|
| 7 |
+
RUN pip install --no-cache-dir -r requirements.txt
|
| 8 |
+
|
| 9 |
+
RUN mkdir /.cache && chmod 777 /.cache
|
| 10 |
+
|
| 11 |
+
RUN mkdir /app/zelda-xl && chmod 777 /app/zelda-xl
|
| 12 |
+
|
| 13 |
+
RUN accelerate config default
|
| 14 |
+
|
| 15 |
+
CMD ["accelerate", "launch", "train.py", "--mixed_precision=fp16", "--learning_rate=1e-5", "--gradient_checkpointing", "--max_train_steps=360", "--lr_scheduler=constant", "--lr_warmup_steps=0", "--use_ema", "--center_crop", "--random_flip", "--pretrained_model_name_or_path=nroggendorff/animexl", "--dataset_name=nroggendorff/zelda", "--resolution=1024", "--train_batch_size=8", "--output_dir=zelda-xl", "--push_to_hub"]
|