Merge pull request #1 from ashawkey/main

fix installing imagedream as a python package
bytedance · Dec 20, 2023 · 13e0556 · 13e0556
2 parents 99dec08 + dcb0c73
commit 13e0556
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 5 deletions.
diff --git 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
diff --git 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