From 61266ab843aed02227fc26e8485210cc2c7c879a Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Mon, 24 Feb 2025 19:37:11 +0000 Subject: [PATCH] adding runtime metrics / system info additional accelerator support, etc. --- src/axolotl/telemetry/manager.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/axolotl/telemetry/manager.py b/src/axolotl/telemetry/manager.py index 301c99c3d..45492d2b9 100644 --- a/src/axolotl/telemetry/manager.py +++ b/src/axolotl/telemetry/manager.py @@ -298,15 +298,13 @@ class TelemetryManager: elif hasattr(torch, "npu") and torch.npu.is_available(): accelerator_type = "npu" for i in range(torch.npu.device_count()): - name = getattr(torch.npu, "get_device_name", lambda x: "NPU")(i) - memory = None if hasattr(torch.npu, "get_device_properties"): memory = torch.npu.get_device_properties(i).total_memory gpu_info.append( { - "name": name, + "name": torch.npu.get_device_name(i), "memory": memory, "type": "npu", }