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