Browse Source

Merge pull request #649 from eclecticc/amd_fix

Fix AMD device capabilities fields
Alex Cheema 5 months ago
parent
commit
263b18a31e
1 changed files with 2 additions and 2 deletions
  1. 2 2
      exo/topology/device_capabilities.py

+ 2 - 2
exo/topology/device_capabilities.py

@@ -211,8 +211,8 @@ async def linux_device_capabilities() -> DeviceCapabilities:
 
     return DeviceCapabilities(
       model="Linux Box ({gpu_name})",
-      chip={gpu_name},
-      memory=gpu_memory_info.total // 2**20,
+      chip=gpu_name,
+      memory=gpu_memory_info // 2**20,
       flops=DeviceFlops(fp32=0, fp16=0, int8=0),
     )