Update app.py
Browse files
app.py
CHANGED
|
@@ -59,9 +59,9 @@ def visualize_cluster(nodes, tp, pp, node_spacing=0.5, gpu_spacing=0.1, nodes_pe
|
|
| 59 |
gpu_i += 1
|
| 60 |
plt.tight_layout()
|
| 61 |
|
| 62 |
-
md = f"## Resulting configuration
|
| 63 |
|
| 64 |
-
TP={tp}, PP={pp}, DP={dp}, where one model instance requires {model_instance} GPUs and {nodes*8-max_gpu_usage} GPUs (in black) cannot be utilized in this configuration. Numbers and color shades indicate pipeline stage if PP>1."
|
| 65 |
|
| 66 |
return md, fig
|
| 67 |
|
|
|
|
| 59 |
gpu_i += 1
|
| 60 |
plt.tight_layout()
|
| 61 |
|
| 62 |
+
md = f"""## Resulting configuration
|
| 63 |
|
| 64 |
+
TP={tp}, PP={pp}, DP={dp}, where one model instance requires {model_instance} GPUs and {nodes*8-max_gpu_usage} GPUs (in black) cannot be utilized in this configuration. Numbers and color shades indicate pipeline stage if PP>1."""
|
| 65 |
|
| 66 |
return md, fig
|
| 67 |
|