def create_configs()

in bench_cluster/create_configs.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

def create_configs(
    out_dir: str,
    model: str,
    gpus: int,
    dp_max: int,
    tp_max: int,
    pp_max: int,
    bapr_max: int,
    gbs_range: tuple[int],
    no_profiler: bool = False,
    cluster: str = "hf",
    exp_name: str = None,
    seq_len: int = 4096,
    recompute_layer: bool = False,
    dry_run: bool = False