Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,10 @@ def visualize_cluster(tp, pp, nodes, nodes_per_row, node_spacing=0.5, gpu_spacin
|
|
61 |
|
62 |
md = f"""## Resulting configuration
|
63 |
|
64 |
-
TP={tp}, PP={pp}, DP={dp}
|
|
|
|
|
|
|
65 |
|
66 |
return md, fig
|
67 |
|
|
|
61 |
|
62 |
md = f"""## Resulting configuration
|
63 |
|
64 |
+
- **3D config: TP={tp}, PP={pp}, DP={dp}**
|
65 |
+
- **one model instance requires {model_instance} GPUs**
|
66 |
+
- **{nodes*8-max_gpu_usage} GPUs (in black) cannot be utilized**
|
67 |
+
- **numbers and color shades indicate pipeline stage if PP>1**"""
|
68 |
|
69 |
return md, fig
|
70 |
|