|
23 | 23 | parser = argparse.ArgumentParser()
|
24 | 24 |
|
25 | 25 | parser.add_argument(
|
26 |
| - "--checkpoint_path", default=None, type=str, required=True, help="Path to the checkpoint to convert." |
| 26 | + "--checkpoint_path", |
| 27 | + default="/Users/sanchitgandhi/convert-audioldm/ldm_trimmed.ckpt", |
| 28 | + type=str, |
| 29 | + required=False, # TODO: revert to True |
| 30 | + help="Path to the checkpoint to convert.", |
27 | 31 | )
|
| 32 | + # !wget https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml |
28 | 33 | parser.add_argument(
|
29 | 34 | "--original_config_file",
|
30 | 35 | default=None,
|
|
43 | 48 | type=str,
|
44 | 49 | help="Type of scheduler to use. Should be one of ['pndm', 'lms', 'ddim', 'euler', 'euler-ancestral', 'dpm']",
|
45 | 50 | )
|
| 51 | + parser.add_argument( |
| 52 | + "--pipeline_type", |
| 53 | + default=None, |
| 54 | + type=str, |
| 55 | + help="The pipeline type. If `None` pipeline will be automatically inferred.", |
| 56 | + ) |
46 | 57 | parser.add_argument(
|
47 | 58 | "--image_size",
|
48 | 59 | default=None,
|
|
57 | 68 | )
|
58 | 69 | parser.add_argument(
|
59 | 70 | "--extract_ema",
|
60 |
| - action="store_true", |
| 71 | + action="store_false", # TODO: revert to store_true |
61 | 72 | help=(
|
62 | 73 | "Only relevant for checkpoints that have both EMA and non-EMA weights. Whether to extract the EMA weights"
|
63 | 74 | " or not. Defaults to `False`. Add `--extract_ema` to extract the EMA weights. EMA weights usually yield"
|
|
74 | 85 | action="store_true",
|
75 | 86 | help="Whether to store pipeline in safetensors format or not.",
|
76 | 87 | )
|
77 |
| - parser.add_argument("--dump_path", default=None, type=str, required=True, help="Path to the output model.") |
| 88 | + parser.add_argument( |
| 89 | + "--dump_path", |
| 90 | + default="/Users/sanchitgandhi/convert-audioldm/diffusers_out_3", |
| 91 | + type=str, |
| 92 | + required=False, # TODO: revert to True |
| 93 | + help="Path to the output model.", |
| 94 | + ) |
78 | 95 | parser.add_argument("--device", type=str, help="Device to use (e.g. cpu, cuda:0, cuda:1, etc.)")
|
79 | 96 | args = parser.parse_args()
|
80 | 97 |
|
|
0 commit comments