Ethanfel/ComfyUI-Tween
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use actual input frame count for all_on_gpu and chunk_size estimates

Browse Source 
Replace hardcoded 199-frame assumption with 2*N-1 from the actual
images input, giving accurate VRAM/RAM estimates for any batch size.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-28 18:56:57 +01:00
parent c4b69321bb
commit 91947c0b8c
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@@ -825,11 +825,11 @@ class VFIOptimizer:
else: else:
batch_size = max(1, min(int(available_mb * 0.85 / per_pair_vram_mb), 64)) batch_size = max(1, min(int(available_mb * 0.85 / per_pair_vram_mb), 64))
# all_on_gpu: estimate output frames for 2x on 2 input frames → 3 output # all_on_gpu: estimate output frames for 2x from actual input count
# More generally, estimate if a modest output (e.g. 100 2x-frames) fits
H, W = images.shape[1], images.shape[2] H, W = images.shape[1], images.shape[2]
N = images.shape[0]
frame_mb = H * W * 3 * 4 / (1024 ** 2) # float32 [C,H,W] frame_mb = H * W * 3 * 4 / (1024 ** 2) # float32 [C,H,W]
estimated_output_frames = 199 # 100 input → 199 output at 2x estimated_output_frames = 2 * N - 1 # 2x is most common multiplier
output_vram_mb = estimated_output_frames * frame_mb output_vram_mb = estimated_output_frames * frame_mb
vram_after_model = total_vram_mb - min_free_mb - model_vram_mb vram_after_model = total_vram_mb - min_free_mb - model_vram_mb
all_on_gpu = output_vram_mb < vram_after_model * 0.5 all_on_gpu = output_vram_mb < vram_after_model * 0.5
@@ -844,8 +844,7 @@ class VFIOptimizer:
# chunk_size: based on system RAM # chunk_size: based on system RAM
system_ram_gb = _get_system_ram_gb() system_ram_gb = _get_system_ram_gb()
system_ram_mb = system_ram_gb * 1024 system_ram_mb = system_ram_gb * 1024
# Estimate if full 2x output fits in 60% of RAM # Check if full 2x output fits in 60% of RAM
# Conservative: assume 100 input frames worth of output
if output_vram_mb < system_ram_mb * 0.6: if output_vram_mb < system_ram_mb * 0.6:
chunk_size = 0 # fits in RAM, no chunking needed chunk_size = 0 # fits in RAM, no chunking needed
else: else: