Spaces:

nroggendorff
/

tune-diffuser

Paused

nroggendorff commited on Jul 19, 2024

Commit

35df8bf

verified ·

1 Parent(s): 6307de2

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -12,4 +12,4 @@ RUN mkdir /app/zelda-xl && chmod 777 /app/zelda-xl
 RUN accelerate config default
-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=4", "--output_dir=zelda-xl", "--push_to_hub"]


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=1", "--output_dir=zelda-xl", "--push_to_hub"]