Skip to content

Commit c1dbbb1

Browse files
authored
Merge pull request #1392 from transformerlab/fix/local-provider-constraints
Fix failing cpu and memory constraints for local provider queue
2 parents 5cf43db + f51d1dd commit c1dbbb1

File tree

1 file changed

+2
-36
lines changed

1 file changed

+2
-36
lines changed

src/renderer/components/Experiment/Tasks/QueueTaskModal.tsx

Lines changed: 2 additions & 36 deletions
Original file line numberDiff line numberDiff line change
@@ -334,42 +334,8 @@ export default function QueueTaskModal({
334334
}
335335
}
336336

337-
// Check CPU requirement
338-
if (taskResources.cpus) {
339-
const requiredCpus = parseInt(String(taskResources.cpus), 10);
340-
const availableCpus = serverInfoData.cpu_count || 0;
341-
if (
342-
!isNaN(requiredCpus) &&
343-
requiredCpus > 0 &&
344-
availableCpus < requiredCpus
345-
) {
346-
issues.push({
347-
type: 'warning',
348-
label: 'CPUs',
349-
required: `${requiredCpus}`,
350-
available: `${availableCpus}`,
351-
});
352-
}
353-
}
354-
355-
// Check memory requirement (task specifies in GB)
356-
if (taskResources.memory) {
357-
const requiredMemoryGB = parseFloat(String(taskResources.memory));
358-
const availableMemoryBytes = serverInfoData.memory?.total || 0;
359-
const availableMemoryGB = availableMemoryBytes / (1024 * 1024 * 1024);
360-
if (
361-
!isNaN(requiredMemoryGB) &&
362-
requiredMemoryGB > 0 &&
363-
availableMemoryGB < requiredMemoryGB
364-
) {
365-
issues.push({
366-
type: 'warning',
367-
label: 'Memory',
368-
required: `${requiredMemoryGB} GB`,
369-
available: `${availableMemoryGB.toFixed(1)} GB`,
370-
});
371-
}
372-
}
337+
// Note: We intentionally skip validating CPU and memory for local providers.
338+
// Only accelerator (GPU) requirements are enforced here.
373339

374340
const hasErrors = issues.some((i) => i.type === 'error');
375341
const hasWarnings = issues.some((i) => i.type === 'warning');

0 commit comments

Comments
 (0)