stzhao commited on
Commit
20f83fb
·
verified ·
1 Parent(s): 417312b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -185,6 +185,8 @@ def plot_parallel_groups(title="Parallel Groups", dp_groups=None, tp_groups=None
185
  def create_interface():
186
  def update_plot(parallel_group_type, tensor_model_parallel_size, pipeline_model_parallel_size, context_parallel_size, unused_text):
187
  world_size = 16 # Fixed world size for 2 machines with 8 GPUs each
 
 
188
 
189
  description = descriptions.get(parallel_group_type, "Invalid parallel group type")
190
 
 
185
  def create_interface():
186
  def update_plot(parallel_group_type, tensor_model_parallel_size, pipeline_model_parallel_size, context_parallel_size, unused_text):
187
  world_size = 16 # Fixed world size for 2 machines with 8 GPUs each
188
+
189
+ print(descriptions)
190
 
191
  description = descriptions.get(parallel_group_type, "Invalid parallel group type")
192