CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-04-22 17:36:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix false vram numbers (#1065)

Browse Source
This commit is contained in:
Harisreedhar
2026-03-23 21:50:48 +05:30
committed by GitHub
parent 0f4a90aebd
commit dbc4b8778e
3 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/system.py
+4 -4
View File
@@ -60,9 +60,9 @@ def detect_nvidia_graphic_devices() -> List[GraphicDevice]:
'value': pynvml.nvmlDeviceGetMemoryInfo(handle).total // (1024 * 1024 * 1024),
'unit': 'GB'
},
'free':
'used':
{
'value': pynvml.nvmlDeviceGetMemoryInfo(handle).free // (1024 * 1024 * 1024),
'value': pynvml.nvmlDeviceGetMemoryInfo(handle).used // (1024 * 1024 * 1024),
'unit': 'GB'
}
},
@@ -131,9 +131,9 @@ def detect_amd_graphic_devices() -> List[GraphicDevice]:
'value': vram_usage.get('vram_total', 0) // (1024 * 1024 * 1024),
'unit': 'GB'
},
'free':
'used':
{
'value': (vram_usage.get('vram_total', 0) - vram_usage.get('vram_used', 0)) // (1024 * 1024 * 1024),
'value': vram_usage.get('vram_used', 0) // (1024 * 1024 * 1024),
'unit': 'GB'
}
},
facefusion/types.py
+1 -1
View File
@@ -292,7 +292,7 @@ ExecutionDeviceProduct = TypedDict('ExecutionDeviceProduct',
ExecutionDeviceVideoMemory = TypedDict('ExecutionDeviceVideoMemory',
{
'total' : Optional[ValueAndUnit],
'free' : Optional[ValueAndUnit]
'used' : Optional[ValueAndUnit]
})
ExecutionDeviceTemperature = TypedDict('ExecutionDeviceTemperature',
{
tests/test_api_metrics.py
+4 -2
View File
@@ -112,9 +112,9 @@ def mock_detect_execution_devices(mocker : MockerFixture) -> None:
'value': 24,
'unit': 'GB'
},
'free':
'used':
{
'value': 20,
'value': 4,
'unit': 'GB'
}
},
@@ -170,6 +170,7 @@ def test_get_metrics(test_client : TestClient) -> None:
assert metrics_body.get('graphic_devices')[0].get('driver_version') == '555.42'
assert metrics_body.get('graphic_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_body.get('graphic_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_body.get('graphic_devices')[0].get('video_memory').get('used').get('value') == 4
assert metrics_body.get('disks')[0].get('total').get('value') == 500
assert metrics_body.get('disks')[0].get('free').get('unit') == 'GB'
@@ -204,6 +205,7 @@ def test_websocket_metrics(test_client : TestClient) -> None:
assert metrics_set.get('graphic_devices')[0].get('driver_version') == '555.42'
assert metrics_set.get('graphic_devices')[0].get('product').get('name') == 'RTX 4090'
assert metrics_set.get('graphic_devices')[0].get('video_memory').get('total').get('value') == 24
assert metrics_set.get('graphic_devices')[0].get('video_memory').get('used').get('value') == 4
assert metrics_set.get('disks')[0].get('total').get('value') == 500
assert metrics_set.get('disks')[0].get('free').get('unit') == 'GB'