bench_cluster/create_configs.py [159:167]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    config_content = deepcopy(base_config)
    config_content["tokens"]["sequence_length"] = seq_len
    config_content["parallelism"]["recompute_layer"] = recompute_layer
    update_config_based_on_model(model, config_content)
    
    if cluster == "hf":
        tp_max_cluster = 8
    elif cluster == "swiss-ai":
        tp_max_cluster = 4 # GH200
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



bench_cluster/create_configs.py [228:236]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    config_content = deepcopy(base_config)
    config_content["tokens"]["sequence_length"] = seq_len
    config_content["parallelism"]["recompute_layer"] = recompute_layer
    update_config_based_on_model(model, config_content)
    
    if cluster == "hf":
        tp_max_cluster = 8
    elif cluster == "swiss-ai":
        tp_max_cluster = 4 # GH200
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -