Remove special VRAM allowance for local workers

- Local workers now require sufficient VRAM like other workers
- Since the server has 24GB VRAM and jobs need 16GB, the check passes normally
parent 837264f3
...@@ -1109,8 +1109,8 @@ class ClusterMaster: ...@@ -1109,8 +1109,8 @@ class ClusterMaster:
total_vram = 0 # CPU has no VRAM limit total_vram = 0 # CPU has no VRAM limit
print(f"DEBUG: CPU-only worker detected") print(f"DEBUG: CPU-only worker detected")
# Check if worker has enough VRAM (skip check for CPU workers or local workers) # Check if worker has enough VRAM (skip check for CPU workers)
has_sufficient_vram = total_vram >= required_vram_gb or not has_gpu or client_id == 'local' has_sufficient_vram = total_vram >= required_vram_gb or not has_gpu
if has_sufficient_vram: if has_sufficient_vram:
available_workers.append((proc_key, client_info['weight'], total_vram)) available_workers.append((proc_key, client_info['weight'], total_vram))
print(f"DEBUG: Worker {proc_key} accepted (VRAM: {total_vram}GB, required: {required_vram_gb}GB)") print(f"DEBUG: Worker {proc_key} accepted (VRAM: {total_vram}GB, required: {required_vram_gb}GB)")
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment