Update app.py

app.py

@@ -171,43 +171,43 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                         value=44, 
                         info="Number of transformer layers used in the model"
                     )          
-        with gr.Column("User Defined"):  
-            gr.Markdown("## User Defined")
-            num_gpus = gr.Number(
-                label="Number of GPUs", 
-                value=1, 
-                info="Number of GPUs used for training"
-            )
-            tensor_parallel_size = gr.Number(
-                label="Tensor Parallel Size", 
-                value=1, 
-                info="Tensor parallel degree (1 if not used)"
-            )
-            pipeline_parallel_size = gr.Number(
-                label="Pipeline Parallel Size", 
-                value=1, 
-                info="Pipeline parallel degree (1 if not used)"
-            )
-            batch_size_per_gpu = gr.Number(
-                label="Batch Size per GPU", 
-                value=8, 
-                info="Batch size per GPU"
-            )
-            ffn_expansion_factor = gr.Number(
-                label="FFN Expansion Factor", 
-                value=4, 
-                info="How much the MLP hidden size expands"
-            )
-            is_mixed_precision = gr.Checkbox(
-                label="Mixed Precision", 
-                value=True, 
-                info="Whether mixed precision is enabled"
-            )
-            misc_mem_gib = gr.Number(
-                label="Miscellaneous Memory Overhead (GiB)", 
-                value=5, 
-                info="Miscellaneous memory overhead per GPU by DL frameworks, communication libraries, etc."
-            )
+            with gr.Column("User Defined"):  
+                gr.Markdown("## User Defined")
+                num_gpus = gr.Number(
+                    label="Number of GPUs", 
+                    value=1, 
+                    info="Number of GPUs used for training"
+                )
+                tensor_parallel_size = gr.Number(
+                    label="Tensor Parallel Size", 
+                    value=1, 
+                    info="Tensor parallel degree (1 if not used)"
+                )
+                pipeline_parallel_size = gr.Number(
+                    label="Pipeline Parallel Size", 
+                    value=1, 
+                    info="Pipeline parallel degree (1 if not used)"
+                )
+                batch_size_per_gpu = gr.Number(
+                    label="Batch Size per GPU", 
+                    value=8, 
+                    info="Batch size per GPU"
+                )
+                ffn_expansion_factor = gr.Number(
+                    label="FFN Expansion Factor", 
+                    value=4, 
+                    info="How much the MLP hidden size expands"
+                )
+                is_mixed_precision = gr.Checkbox(
+                    label="Mixed Precision", 
+                    value=True, 
+                    info="Whether mixed precision is enabled"
+                )
+                misc_mem_gib = gr.Number(
+                    label="Miscellaneous Memory Overhead (GiB)", 
+                    value=5, 
+                    info="Miscellaneous memory overhead per GPU by DL frameworks, communication libraries, etc."
+                )
 
         calc_memory_button = gr.Button("Calculate Memory")
         memory_result = gr.Textbox(label="Memory Calculation Result", interactive=False)