Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in
Toggle navigation
V
vidai
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
SexHackMe
vidai
Commits
3790f911
Commit
3790f911
authored
Oct 09, 2025
by
Stefy Lanza (nextime / spora )
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Fix VRAM overhead calculation - use only model base VRAM + configurable overhead, not +6GB extra
parent
7725cb9b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
6 deletions
+5
-6
cluster_master.py
vidai/cluster_master.py
+5
-6
No files found.
vidai/cluster_master.py
View file @
3790f911
...
...
@@ -1119,7 +1119,7 @@ class ClusterMaster:
# Step 1: Determine VRAM required for the model (includes overhead)
required_vram_gb
=
estimate_model_vram_requirements
(
model_path
)
print
(
f
"DEBUG: Required VRAM: {required_vram_gb}GB (includes overhead)"
)
print
(
f
"DEBUG: Required VRAM: {required_vram_gb}GB (includes
model
overhead)"
)
# Step 2: Determine workers with sufficient GPU memory
available_workers
=
[]
...
...
@@ -1184,14 +1184,13 @@ class ClusterMaster:
print
(
f
"DEBUG: CPU-only worker detected"
)
# Check if worker has enough VRAM (skip check for CPU workers)
# Worker must have at least 6GB more VRAM than required
required_vram_with_buffer
=
required_vram_gb
+
6
if
has_gpu
else
required_vram_gb
has_sufficient_vram
=
total_vram
>=
required_vram_with_buffer
or
not
has_gpu
# Required VRAM already includes model-specific overhead
has_sufficient_vram
=
total_vram
>=
required_vram_gb
or
not
has_gpu
if
has_sufficient_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_
with_buffer
}GB)"
)
print
(
f
"DEBUG: Worker {proc_key} accepted (VRAM: {total_vram}GB, required: {required_vram_
gb
}GB)"
)
else
:
print
(
f
"DEBUG: Worker {proc_key} rejected - insufficient VRAM ({total_vram}GB < {required_vram_
with_buffer
}GB)"
)
print
(
f
"DEBUG: Worker {proc_key} rejected - insufficient VRAM ({total_vram}GB < {required_vram_
gb
}GB)"
)
if
not
available_workers
:
return
None
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment