.

app.py

@@ -106,7 +106,7 @@ with gr.Blocks() as demo:
                     tp = gr.Number(1, label="Tensor Parallelism")
                     pp = gr.Number(1, label="Pipeline Parallelism")
                     dp = gr.Number(1, label="Data Parallelism")
-                    zero_stage = gr.Radio([0, 1, 2, 3], value=0, label="ZeRO Stage")
+                    zero_stage = gr.Radio([0, 1], value=0, label="ZeRO Stage")
 
                 manual_submit = gr.Button("Calculate Memory (Manual Input)")