Spaces:

lvwerra
/

3d

Running

lvwerra HF staff commited on Jun 19, 2024

Commit

98a83bc

verified ·

1 Parent(s): e0e4e61

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import math
 plt.rcParams['axes.prop_cycle'] = plt.cycler(color=["#FF6600","#FFBE00","#496767", "#87A19E",  "#FF9200",  "#0F3538","#F8E08E", "#0F2021","#FAFAF0"])
-def visualize_cluster(nodes, tp, pp, node_spacing=0.5, gpu_spacing=0.1, nodes_per_row=8):
     gpus_per_row = 2
     gpus_per_column = 4

 plt.rcParams['axes.prop_cycle'] = plt.cycler(color=["#FF6600","#FFBE00","#496767", "#87A19E",  "#FF9200",  "#0F3538","#F8E08E", "#0F2021","#FAFAF0"])
+def visualize_cluster(tp, pp, nodes, nodes_per_row, node_spacing=0.5, gpu_spacing=0.1 ):
     gpus_per_row = 2
     gpus_per_column = 4