fix dtype inspect

Signed-off-by: Vladimir Mandic <mandic00@live.com>
2026-09-19 09:14:35 +02:00 · 2026-09-11 10:24:34 +02:00
parent 3f7a494b81
commit c3fa079aad
3 changed files with 8 additions and 3 deletions
@@ -174,7 +174,10 @@ def report_model_stats(module_name, module):
            size, _params = get_module_size(module)
        quant = getattr(module, "quantization_method", None)
        params = sum(p.numel() for p in module.parameters(recurse=True))
-        dtype = next(module.parameters()).dtype if any(module.parameters()) else None
+        try:
+            dtype = next(module.parameters(), torch.tensor([])).dtype
+        except Exception:
+            dtype = None
        logical = get_logical_param_count(module)
        log.debug(f'Module: name={module_name} cls={module.__class__.__name__} size={size:.3f} params={params} logical={logical} quant={quant} dtype={dtype}')
    except Exception as e: