Koke_Cacao commited on
Commit
9db01cb
1 Parent(s): 73f5200

:bug: fix arguments

Browse files
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()