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

Fix total_mem → total_memory attribute on CudaDeviceProperties

Browse Source 
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-28 18:33:45 +01:00
parent d2e7db49c7
commit 27c5bcf362
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@@ -815,7 +815,7 @@ class VFIOptimizer:
per_pair_vram_mb = max(per_pair_vram_bytes / (1024 ** 2), 1.0) per_pair_vram_mb = max(per_pair_vram_bytes / (1024 ** 2), 1.0)
# --- Compute settings --- # --- Compute settings ---
total_vram_mb = torch.cuda.get_device_properties(device).total_mem / (1024 ** 2) total_vram_mb = torch.cuda.get_device_properties(device).total_memory / (1024 ** 2)
min_free_mb = min_free_vram_gb * 1024 min_free_mb = min_free_vram_gb * 1024
available_mb = total_vram_mb - min_free_mb - model_vram_mb available_mb = total_vram_mb - min_free_mb - model_vram_mb