Update app.py
Browse files
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 |
|