wregtgh

app.py

@@ -543,8 +543,16 @@ with gr.Blocks(title="Text-to-SQL RLHF") as demo:
             
             with gr.Accordion("Task 1: Parallel Reward Benchmark", open=False):
                 gr.Markdown("*(Simulates the heavy RLHF training workload by running hundreds of complex SQL queries concurrently to test SQLite multi-threading performance.)*")
+            #    *******************
+                core_count = os.cpu_count() or 2
+                smart_worker_default = min(32, core_count * 2) 
+
                 t1_n = gr.Number(value=20, precision=0, label="Rollouts (n)")
-                t1_workers = gr.Number(value=10, precision=0, label="Max workers")
+                t1_workers = gr.Number(value=smart_worker_default, precision=0, label=f"Max workers (Auto-detected: {core_count} cores)")
+                #****************
+                # t1_n = gr.Number(value=20, precision=0, label="Rollouts (n)")
+
+                # t1_workers = gr.Number(value=10, precision=0, label="Max workers")
                 t1_run = gr.Button("Run Task 1 benchmark")
                 t1_out = gr.Textbox(label="Output", lines=12)
                 t1_plot = gr.HTML(label="Plot (if generated)")