diff --git a/python_coreml_stable_diffusion/mixed_bit_compression_apply.py b/python_coreml_stable_diffusion/mixed_bit_compression_apply.py index b65f07cd..8e2f8b13 100644 --- a/python_coreml_stable_diffusion/mixed_bit_compression_apply.py +++ b/python_coreml_stable_diffusion/mixed_bit_compression_apply.py @@ -120,6 +120,10 @@ def get_tensor_hash(tensor): "If specified, the specified VAE will be converted instead of the one associated to the `--model-version` checkpoint. " "No precision override is applied when using a custom VAE." )) + parser.add_argument( + "--sd3-version", + action="store_true", + help=("If specified, the pre-trained model will be treated as an SD3 model.")) args = parser.parse_args() diff --git a/python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py b/python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py index 505895c7..e1b6e07a 100644 --- a/python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py +++ b/python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py @@ -578,6 +578,11 @@ def main(args): "If specified, the specified VAE will be converted instead of the one associated to the `--model-version` checkpoint. " "No precision override is applied when using a custom VAE." )) + # needed since this calls `torch2coreml` and that would throw an error + parser.add_argument( + "--sd3-version", + action="store_true", + help=("If specified, the pre-trained model will be treated as an SD3 model.")) args = parser.parse_args() main(args)