--- library_name: diffusers --- # Model Card for Model ID House plans #trained on:WhiteCase ## Model Details ```python class TrainingConfig: image_size = 256 # the generated image resolution train_batch_size = 4 eval_batch_size = 4 # how many images to sample during evaluation num_epochs = 300 gradient_accumulation_steps = 1 learning_rate = 1e-4 lr_warmup_steps = 500 save_image_epochs = 10 save_model_epochs = 30 mixed_precision = 'fp16' # `no` for float32, `fp16` for automatic mixed precision output_dir = 'ddpm-butterflies-128' # the model namy locally and on the HF Hub push_to_hub = True # whether to upload the saved model to the HF Hub hub_private_repo = False overwrite_output_dir = True # overwrite the old model when re-running the notebook seed = 0 config = TrainingConfig() ``` ### Model Description