From e9df4444bebcf241118faf018c214323596eac62 Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Mon, 24 Feb 2025 20:02:16 +0000 Subject: [PATCH] remove duplicate info --- src/axolotl/telemetry/manager.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/src/axolotl/telemetry/manager.py b/src/axolotl/telemetry/manager.py index 45492d2b9..4a61baf36 100644 --- a/src/axolotl/telemetry/manager.py +++ b/src/axolotl/telemetry/manager.py @@ -248,7 +248,6 @@ class TelemetryManager: { "name": torch.cuda.get_device_name(i), "memory": torch.cuda.get_device_properties(i).total_memory, - "type": "cuda", } ) @@ -262,7 +261,6 @@ class TelemetryManager: "memory": torch.hip.get_device_properties(i).total_memory if hasattr(torch.hip, "get_device_properties") else None, - "type": "hip", } ) @@ -274,7 +272,6 @@ class TelemetryManager: "name": "Apple Silicon", # NOTE: this is memory allocated to this process, not total memory "memory": torch.mps.driver_allocated_memory(), - "type": "mps", } ) @@ -290,7 +287,6 @@ class TelemetryManager: { "name": torch.xpu.get_device_name(i), "memory": memory, - "type": "xpu", } ) @@ -306,7 +302,6 @@ class TelemetryManager: { "name": torch.npu.get_device_name(i), "memory": memory, - "type": "npu", } )