feat: Detect and log VRAM allocation mode

2025-08-26 00:14:37 -05:00
parent 07df43b863
commit b351c5dbc3
1 changed files with 8 additions and 0 deletions
@@ -142,6 +142,14 @@ def analyze_safetensor_loading(model_patcher, allocations_str):
        if not distorch_alloc:
            distorch_alloc = calculate_safetensor_vvram_allocation(model_patcher, virtual_vram_str)

+    # Determine the allocation mode
+    mode = "fraction"
+    if any(c in distorch_alloc.lower() for c in ['g', 'm', 'k', 'b']):
+        mode = "byte"
+    elif "%" in distorch_alloc:
+        mode = "ratio"
+    logger.info(f"[MultiGPU_DisTorch2] Detected allocation mode: {mode}")
+
    eq_line = "=" * 50
    dash_line = "-" * 50
    fmt_assign = "{:<18}{:>7}{:>14}{:>10}"