From 593d0f8e4f1d9d913b4933b0622146451ad6039d Mon Sep 17 00:00:00 2001
From: Arjun Gupta <arjuntheprogrammer@gmail.com>
Date: Sat, 28 Jun 2025 21:21:18 +0530
Subject: [PATCH] Updated README - added bedroom checkpoints

---
 README.md | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/README.md b/README.md
index 6aea256..dd92163 100644
--- a/README.md
+++ b/README.md
@@ -63,6 +63,11 @@ python3 src/train_edit.py --config_file configs/bedroom_sg_diffusion.yaml --outp
 
 You can change `config_file` to other room types. All config files are under [configs](./configs/) folder.
 
+Pretained checkpoints download links for bedroom scene (batch_size used: 256):
+
+- SG: <https://huggingface.co/auraml/EditRoomOrg/resolve/main/weights/threedfront_bedroom_sg_model-epoch%3D213-val_loss%3D0.107.ckpt>
+- SG2SC: <https://huggingface.co/auraml/EditRoomOrg/resolve/main/weights/threedfront_bedroom_sg2sc_model-epoch%3D273-val_loss%3D0.018.ckpt>
+
 ### 4. Evaluation
 To run the evaluation, we need both `SG_WEIGHT` and `SG2SC_WEIGHT`. Those weights should be found under `output_directory` for training.