lvwerra HF staff commited on
Commit
e0e4e61
·
verified ·
1 Parent(s): 1c2f6af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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