Function get_args

train_code/train_cogvideox_motion.py:86–108

()

################################################################################################################################
def get_args():
    parser = argparse.ArgumentParser(description="Simple example of a ControlNet training script.")
    parser.add_argument(
                            "--config_path",
                            type=str,
                            default="config/train_cogvideox_motion.yaml",
                            help="Path to the config.",
                        )
    parser.add_argument(
                            "--use_8BitAdam",
                            type=bool,
                            default = False,
                            help = "Whether we use the 8BitAdam.",
                        )
    parser.add_argument(
                            "--debug",
                            type = bool,
                            default = False,
                            help = "Whether we are in the debug model: don't use torch.compile",
                        )

    args = parser.parse_args()
    return args