File size: 480 Bytes
1629854 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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="constant"
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 |