diff --git a/extern/ImageDream/README.md b/extern/ImageDream/README.md
index 3db63a7..afa82a9 100644
--- a/extern/ImageDream/README.md
+++ b/extern/ImageDream/README.md
@@ -14,6 +14,11 @@ Setup environment as in [Stable-Diffusion](https://github.com/Stability-AI/stabl
 pip install -r requirements.txt
 ```
 
+To use ImageDream as a python module, you can install it by `pip install -e .` or:
+```bash
+pip install git+https://github.com/bytedance/ImageDream/#subdirectory=extern/ImageDream
+```
+
 ## Image-to-Multi-View
 Clone the modelcard on the [Huggingface ImageDream Model Page](https://huggingface.co/Peng-Wang/ImageDream/) under ```./release_models/```
 
diff --git a/extern/ImageDream/astronaut_local_dream.png b/extern/ImageDream/astronaut_local_dream.png
deleted file mode 100644
index 7b9753e..0000000
Binary files a/extern/ImageDream/astronaut_local_dream.png and /dev/null differ
diff --git a/extern/ImageDream/imagedream/model_zoo.py b/extern/ImageDream/imagedream/model_zoo.py
index 0ed867a..d65529a 100644
--- a/extern/ImageDream/imagedream/model_zoo.py
+++ b/extern/ImageDream/imagedream/model_zoo.py
@@ -11,13 +11,13 @@
 
 PRETRAINED_MODELS = {
     "sd-v2.1-base-4view-ipmv": {
-        "config": "sd-v2-base-ipmv.yaml",
-        "repo_id": "ImageDream/ImageDream",
+        "config": "sd_v2_base_ipmv.yaml",
+        "repo_id": "Peng-Wang/ImageDream",
         "filename": "sd-v2.1-base-4view-ipmv.pt",
     },
     "sd-v2.1-base-4view-ipmv-local": {
-        "config": "sd-v2-base-ipmv-local.yaml",
-        "repo_id": "ImageDream/ImageDream",
+        "config": "sd_v2_base_ipmv_local.yaml",
+        "repo_id": "Peng-Wang/ImageDream",
         "filename": "sd-v2.1-base-4view-ipmv-local.pt",
     },
 }
diff --git a/extern/ImageDream/release_models b/extern/ImageDream/release_models
deleted file mode 120000
index dcb6ef2..0000000
--- a/extern/ImageDream/release_models
+++ /dev/null
@@ -1 +0,0 @@
-/mnt/bn/wp-maliva-bytenas/mlx/users/peng.wang/playground/repo/dalle2/release_models
\ No newline at end of file