update nvml

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-02-01 10:36:12 -05:00
parent 19b01e5000
commit ceaf0238fa
3 changed files with 7 additions and 1 deletions
+3
View File
@@ -25,6 +25,9 @@
- **prompt parser**: log stats with tokens, sections and min/avg/max weights
- **prompt parser**: add setting to ignore line breaks in prompt
set in *settings -> prompt settings -> use line breaks*
- **nvml cli**: run nvidia-management-lib interrogate from cli
already available in ui in generate -> right click -> nvidia
> python modules/api/nvml.py
- **Refactor**:
- unified trace handler with configurable tracebacks
- **Fixes**:
+3
View File
@@ -86,6 +86,9 @@ def get_nvml():
if __name__ == '__main__':
nvml_initialized = True
import pynvml # pylint: disable=redefined-outer-name
pynvml.nvmlInit()
from rich import print as rprint
for gpu in get_nvml():
rprint(gpu)