Update README.md

README.md

@@ -5,11 +5,30 @@ library_name: diffusers
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-
+House plan sketches
 
 
 ## Model Details
 
+class TrainingConfig:
+    image_size = 192  # the generated image resolution
+    train_batch_size = 8
+    eval_batch_size = 8  # how many images to sample during evaluation
+    num_epochs = 200
+    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 = False  # whether to upload the saved model to the HF Hub
+    hub_private_repo = False  
+    overwrite_output_dir = False  # overwrite the old model when re-running the notebook
+    seed = 0
+
+config = TrainingConfig()
+
 ### Model Description
 
 <!-- Provide a longer summary of what this model is. -->