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 |
|