CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-06 02:27:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

rename video memory to memory, add amdsmi library

Browse Source
This commit is contained in:
henryruhs
2026-05-08 12:51:51 +02:00
parent 74a9df35a6
commit 9b878431d3
6 changed files with 238 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_metrics.py
+5 -5
View File
@@ -105,7 +105,7 @@ def mock_detect_execution_devices(mocker : MockerFixture) -> None:
'vendor': 'NVIDIA',
'name': 'RTX 4090'
},
'video_memory':
'memory':
{
'total':
{
@@ -169,8 +169,8 @@ 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('graphic_devices')[0].get('memory').get('total').get('value') == 24
assert metrics_body.get('graphic_devices')[0].get('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'
@@ -221,8 +221,8 @@ 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('graphic_devices')[0].get('memory').get('total').get('value') == 24
assert metrics_set.get('graphic_devices')[0].get('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'