if __name__ == "__main__":
    import deepspeed  # noqa: F401