From b351c5dbc3aaca4d5cc49cde25edee02ff372f66 Mon Sep 17 00:00:00 2001
From: John Pollock <pollockjj@gmail.com>
Date: Tue, 26 Aug 2025 00:14:37 -0500
Subject: [PATCH] feat: Detect and log VRAM allocation mode

---
 distorch_2.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/distorch_2.py b/distorch_2.py
index e4788a9..50c3f9c 100644
--- a/distorch_2.py
+++ b/distorch_2.py
@@ -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}"