finalize remote weights

README.md

@@ -2,14 +2,10 @@
 
 A **unified** diffusers implementation of [MVDream](https://github.com/bytedance/MVDream) and [ImageDream](https://github.com/bytedance/ImageDream).
 
-We provide converted `fp16` weights on [huggingface](TODO).
+We provide converted `fp16` weights on huggingface:
+* [MVDream](https://huggingface.co/ashawkey/mvdream-sd2.1-diffusers)
+* [ImageDream](https://huggingface.co/ashawkey/imagedream-ipmv-diffusers)
 
-### Usage
-
-```bash
-python run_mvdream.py "a cute owl"
-python run_imagedream.py data/anya_rgba.png
-```
 
 ### Install
 ```bash
@@ -17,6 +13,13 @@ python run_imagedream.py data/anya_rgba.png
 pip install -r requirements.txt
 ```
 
+### Usage
+
+```bash
+python run_mvdream.py "a cute owl"
+python run_imagedream.py data/anya_rgba.png
+```
+
 ### Convert weights
 
 MVDream:

run_imagedream.py

@@ -5,8 +5,8 @@ import argparse
 from pipeline_mvdream import MVDreamPipeline
 
 pipe = MVDreamPipeline.from_pretrained(
-    "./weights_imagedream", # local weights
-    # "ashawkey/mvdream-sd2.1-diffusers",
+    # "./weights_imagedream", # local weights
+    "ashawkey/imagedream-ipmv-diffusers", # remote weights
     torch_dtype=torch.float16,
     trust_remote_code=True,
 )

run_mvdream.py

@@ -5,8 +5,8 @@ import argparse
 from pipeline_mvdream import MVDreamPipeline
 
 pipe = MVDreamPipeline.from_pretrained(
-    "./weights_mvdream", # local weights
-    # 'ashawkey/mvdream-sd2.1-diffusers', # remote is not 
+    # "./weights_mvdream", # local weights
+    'ashawkey/mvdream-sd2.1-diffusers', # remote weights
     torch_dtype=torch.float16,
     trust_remote_code=True,
 )