def create_single_config()

in bench_cluster/create_configs.py [0:0]

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

def create_single_config(
    out_dir: str,
    model: str,
    gpus: int,
    dp: int,
    tp: int,
    pp: int,
    bapr: int,
    mbs: int,
    no_profiler: bool = False,
    cluster: str = "hf",
    exp_name: str = None,
    seq_len: int = 4096,
    recompute_layer: bool = False,
    dry_run: bool = False