File size: 582 Bytes
011b70c dea9f51 011b70c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
args = Namespace(
image_path="/content/images/Training Images",
prompt_path="/content/data.csv",
output_path="/content/output",
model="CompVis/stable-diffusion-v1-4",
seed=1703,
mixed_precision="bf16",
rank=4,
use_rslora=True,
init_lora_weights="gaussian",
lr=1e-5,
gradient_accumulation_steps=1,
train_batch_size=64,
num_train_epochs=20,
lr_scheduler="constant",
lr_warmup_steps=1,
snr_gamma=5,
size=268,
max_grad_norm=1,
use_8bit_adam=True,
checkpointing_steps=1100,
resume_from_checkpoint=None
) |