| import subprocess |
| subprocess.run(["accelerate", "launch", "--mixed_precision=fp16", "train_text_to_image_lora.py", |
| "--pretrained_model_name_or_path=runwayml/stable-diffusion-v1-5", |
| "--dataset_name=pcuenq/oxford-pets", |
| "--dataloader_num_workers=8", |
| "--resolution=512", "--center_crop", "--random_flip", |
| "--train_batch_size=1", |
| "--gradient_accumulation_steps=4", |
| "--max_train_steps=5000", |
| "--learning_rate=1e-04", |
| "--max_grad_norm=1", |
| "--lr_scheduler=cosine", "--lr_warmup_steps=0", |
| "--output_dir=/home/htelvis92/182/", |
| "--push_to_hub", |
| "--hub_model_id=pets", |
| "--checkpointing_steps=500", |
| "--validation_prompt=Totoro", |
| "--seed=1337", |
| "--caption_column=label"]) |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
