Fix total_mem → total_memory attribute on CudaDeviceProperties
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
1 parent
d2e7db49c7
commit
27c5bcf362
1 file changed
+1
-1
@@ -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
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user