@@ -24,9 +24,9 @@ def __init__(self):
24
24
# Hardware override section
25
25
hw_layout = QHBoxLayout ()
26
26
27
- vram , bandwidth , num_gpus = LLMcalc .get_vram_specs ()
27
+ vram , bandwidth , num_gpus , brand = LLMcalc .get_vram_specs ()
28
28
29
- if not vram : vram = 0
29
+ if not vram : vram = 0 # Just in case
30
30
if not bandwidth : bandwidth = 0
31
31
32
32
vram_label = QLabel ("VRAM (GB):" )
@@ -65,15 +65,15 @@ def __init__(self):
65
65
def update_system_info (self , real_bandwidth = None , real_num_gpus = None ):
66
66
"""Update system information display"""
67
67
total_ram = LLMcalc .get_ram_specs ()
68
- vram , bandwidth , num_gpus = LLMcalc .get_vram_specs ()
68
+ vram , bandwidth , num_gpus , brand = LLMcalc .get_vram_specs ()
69
69
ram_bandwidth = LLMcalc .get_memory_bandwidth ()
70
70
71
71
if real_num_gpus : num_gpus = real_num_gpus
72
72
if real_bandwidth : bandwidth = real_bandwidth
73
73
if not vram : vram = 0
74
74
if not bandwidth : bandwidth = 0
75
75
76
- info = f"System Info: RAM { total_ram :.2f} GB | VRAM { num_gpus } x{ vram :.2f} GB | GPU BW ~{ bandwidth :.1f} GB/s | RAM BW { ram_bandwidth :.2f } GB/s "
76
+ info = f"System Info: RAM { total_ram :.2f} GB | RAM BW { ram_bandwidth :.2f } GB/s | VRAM { num_gpus } x{ vram :.2f} GB | GPU BW ~{ bandwidth :.1f} GB/s | GPU: { brand } "
77
77
self .system_info .setText (info )
78
78
79
79
def calculate (self ):
@@ -93,7 +93,7 @@ def calculate(self):
93
93
94
94
params_b = LLMcalc .convert_params_to_b (params_text )
95
95
total_ram = LLMcalc .get_ram_specs ()
96
- vram , bandwidth , num_gpus = LLMcalc .get_vram_specs ()
96
+ vram , bandwidth , num_gpus , brand = LLMcalc .get_vram_specs ()
97
97
ram_bandwidth = LLMcalc .get_memory_bandwidth ()
98
98
99
99
# Apply overrides
0 commit comments