| instance_prompt="a sks dog" | |
| class_prompt="a dog" | |
| num_inference_steps=50 | |
| guidance_scale=7.5 | |
| train_batch_size=1 | |
| learning_rate="2e-6" | |
| lr_scheduler="linear" | |
| model_id_finetuned = "/content/model" | |
| model_id_pretrained = "stabilityai/stable-diffusion-2-base" | |
| num_class_images=200 | |
| train_steps=600 | |
| sample_output_dir="/content/samples" | |
| model_output_dir="/content/model" | |
| class_data_dir="/content/resized_class" | |
| instance_data_dir="/content/resized_instance" | |
| image_size=512 |