remove duplicate info

2025-02-24 20:02:16 +00:00
parent ffd2985750
commit e9df4444be
1 changed files with 0 additions and 5 deletions
--- 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",
                    }
                )