Fix total_mem → total_memory attribute on CudaDeviceProperties

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-02-28 18:33:45 +01:00
parent d2e7db49c7
commit 27c5bcf362

View File

@@ -815,7 +815,7 @@ class VFIOptimizer:
per_pair_vram_mb = max(per_pair_vram_bytes / (1024 ** 2), 1.0)
# --- 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
available_mb = total_vram_mb - min_free_mb - model_vram_mb