Fix cluster nodes modal to show per-worker driver selects

- Show individual select forms for every worker in the driver modal
- Update API to handle per-worker driver selection for local nodes
- Maintain compatibility with existing backend switching logic

templates/admin/cluster_nodes.html

@@ -226,37 +226,6 @@ function openDriverModal(hostname, token, hostnameValue) {
     const container = document.getElementById('workerDriverSelections');
     container.innerHTML = '';
 
-    // Handle local vs remote nodes differently
-    if (node.is_local) {
-        // For local node, use single driver selection for all workers
-        const availableGpuBackends = node.available_backends.filter(b => b !== 'cpu');
-
-        if (availableGpuBackends.length === 0) {
-            container.innerHTML = '<p>No GPU backends available for local workers.</p>';
-            return;
-        }
-
-        let optionsHtml = '';
-        availableGpuBackends.forEach(backend => {
-            optionsHtml += `<option value="${backend}">${backend.toUpperCase()}</option>`;
-        });
-
-        container.innerHTML = `
-            <div class="form-group">
-                <label for="driver">Driver for all local workers:</label>
-                <select id="driver" name="driver">
-                    ${optionsHtml}
-                </select>
-            </div>
-        `;
-
-        // Set current preference or default to first available
-        const select = document.getElementById('driver');
-        // For local, default to the first GPU backend available
-        select.value = availableGpuBackends[0];
-
-    } else {
-        // For remote nodes, show worker-specific selections
     // Get GPU-requiring workers (analysis_cuda, training_cuda, analysis_rocm, training_rocm)
     const gpuWorkers = [];
     if (node.workers && node.workers.length > 0) {
@@ -313,7 +282,6 @@ function openDriverModal(hostname, token, hostnameValue) {
             select.value = availableGpuBackends[0];
         }
     });
-    }
 
     document.getElementById('driverModal').style.display = 'block';
 }

vidai/web.py

@@ -730,10 +730,15 @@ def api_set_client_driver():
 
     # Handle local workers
     if token == 'local':
-        # For local workers, expect a single driver for all workers
-        driver = request.form.get('driver')
-        if not driver or driver not in ['cuda', 'rocm', 'cpu']:
-            return {'success': False, 'error': 'Invalid driver - only CUDA, ROCm, and CPU are supported'}, 400
+        # For local workers, parse worker-specific drivers but use the same backend for all
+        driver = None
+        for key, value in request.form.items():
+            if key.startswith('worker_') and key.endswith('_driver'):
+                if value in ['cuda', 'rocm', 'cpu']:
+                    driver = value
+                    break  # Use the first valid driver found
+        if not driver:
+            return {'success': False, 'error': 'No valid driver found for local workers'}, 400
         success = switch_local_worker_backends(driver)
         return {'success': success}