Koke_Cacao
commited on
Commit
•
9db01cb
1
Parent(s):
73f5200
:bug: fix arguments
Browse files- README.md +1 -0
- scripts/convert_mvdream_to_diffusers.py +1 -1
README.md
CHANGED
@@ -34,4 +34,5 @@ wget https://raw.githubusercontent.com/bytedance/MVDream/main/mvdream/configs/sd
|
|
34 |
Hugging Face diffusers weights are converted by script:
|
35 |
```bash
|
36 |
python ./scripts/convert_mvdream_to_diffusers.py --checkpoint_path ./sd-v1.5-4view.pt --dump_path . --original_config_file ./sd-v1.yaml --test
|
|
|
37 |
```
|
|
|
34 |
Hugging Face diffusers weights are converted by script:
|
35 |
```bash
|
36 |
python ./scripts/convert_mvdream_to_diffusers.py --checkpoint_path ./sd-v1.5-4view.pt --dump_path . --original_config_file ./sd-v1.yaml --test
|
37 |
+
python ./scripts/convert_mvdream_to_diffusers.py --checkpoint_path ./sd-v2.1-base-4view.pt --dump_path . --original_config_file ./sd-v2-base.yaml --test
|
38 |
```
|
scripts/convert_mvdream_to_diffusers.py
CHANGED
@@ -739,7 +739,7 @@ if __name__ == "__main__":
|
|
739 |
help="Whether to store pipeline in safetensors format or not.",
|
740 |
)
|
741 |
parser.add_argument("--half", action="store_true", help="Save weights in half precision.")
|
742 |
-
parser.add_argument("--test", help="Whether to test inference after convertion.")
|
743 |
parser.add_argument("--dump_path", default=None, type=str, required=True, help="Path to the output model.")
|
744 |
parser.add_argument("--device", type=str, help="Device to use (e.g. cpu, cuda:0, cuda:1, etc.)")
|
745 |
args = parser.parse_args()
|
|
|
739 |
help="Whether to store pipeline in safetensors format or not.",
|
740 |
)
|
741 |
parser.add_argument("--half", action="store_true", help="Save weights in half precision.")
|
742 |
+
parser.add_argument("--test", action="store_true", help="Whether to test inference after convertion.")
|
743 |
parser.add_argument("--dump_path", default=None, type=str, required=True, help="Path to the output model.")
|
744 |
parser.add_argument("--device", type=str, help="Device to use (e.g. cpu, cuda:0, cuda:1, etc.)")
|
745 |
args = parser.parse_args()
|