()
| 925 | |
| 926 | |
| 927 | def get_args(): |
| 928 | parser = argparse.ArgumentParser() |
| 929 | |
| 930 | def none_or_str(value: str): |
| 931 | if isinstance(value, str) and value.lower() == "none": |
| 932 | return None |
| 933 | return value |
| 934 | |
| 935 | parser.add_argument( |
| 936 | "--original_state_dict_repo_id", |
| 937 | default="Lightricks/LTX-2", |
| 938 | type=none_or_str, |
| 939 | help="HF Hub repo id with LTX 2.0 checkpoint", |
| 940 | ) |
| 941 | parser.add_argument( |
| 942 | "--checkpoint_path", |
| 943 | default=None, |
| 944 | type=str, |
| 945 | help="Local checkpoint path for LTX 2.0. Will be used if `original_state_dict_repo_id` is not specified.", |
| 946 | ) |
| 947 | parser.add_argument( |
| 948 | "--version", |
| 949 | type=str, |
| 950 | default="2.0", |
| 951 | choices=["test", "2.0", "2.3"], |
| 952 | help="Version of the LTX 2.0 model", |
| 953 | ) |
| 954 | |
| 955 | parser.add_argument( |
| 956 | "--combined_filename", |
| 957 | default="ltx-2-19b-dev.safetensors", |
| 958 | type=none_or_str, |
| 959 | help="Filename for combined checkpoint with all LTX 2.0 models (VAE, DiT, etc.)", |
| 960 | ) |
| 961 | parser.add_argument("--vae_prefix", default="vae.", type=str) |
| 962 | parser.add_argument("--audio_vae_prefix", default="audio_vae.", type=str) |
| 963 | parser.add_argument("--dit_prefix", default="model.diffusion_model.", type=str) |
| 964 | parser.add_argument("--vocoder_prefix", default="vocoder.", type=str) |
| 965 | |
| 966 | parser.add_argument("--vae_filename", default=None, type=str, help="VAE filename; overrides combined ckpt if set") |
| 967 | parser.add_argument( |
| 968 | "--audio_vae_filename", default=None, type=str, help="Audio VAE filename; overrides combined ckpt if set" |
| 969 | ) |
| 970 | parser.add_argument("--dit_filename", default=None, type=str, help="DiT filename; overrides combined ckpt if set") |
| 971 | parser.add_argument( |
| 972 | "--vocoder_filename", default=None, type=str, help="Vocoder filename; overrides combined ckpt if set" |
| 973 | ) |
| 974 | parser.add_argument( |
| 975 | "--text_encoder_model_id", |
| 976 | default="google/gemma-3-12b-it-qat-q4_0-unquantized", |
| 977 | type=none_or_str, |
| 978 | help="HF Hub id for the LTX 2.0 base text encoder model", |
| 979 | ) |
| 980 | parser.add_argument( |
| 981 | "--tokenizer_id", |
| 982 | default="google/gemma-3-12b-it-qat-q4_0-unquantized", |
| 983 | type=none_or_str, |
| 984 | help="HF Hub id for the LTX 2.0 text tokenizer", |
no outgoing calls
no test coverage detected
searching dependent graphs…