From ceaf0238fa0f2d77ed73195132654d9f04b2c4fa Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 1 Feb 2025 10:36:12 -0500 Subject: [PATCH] update nvml Signed-off-by: Vladimir Mandic --- CHANGELOG.md | 3 +++ extensions-builtin/sdnext-modernui | 2 +- modules/api/nvml.py | 3 +++ 3 files changed, 7 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 575c0a985..ccfe41ace 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -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**: diff --git a/extensions-builtin/sdnext-modernui b/extensions-builtin/sdnext-modernui index e0b3e5918..5eb85e047 160000 --- a/extensions-builtin/sdnext-modernui +++ b/extensions-builtin/sdnext-modernui @@ -1 +1 @@ -Subproject commit e0b3e5918f49d090c8c340ca44e2c9e6d5bf5127 +Subproject commit 5eb85e047cecfd17d02210a1e77535e27d43cd1e diff --git a/modules/api/nvml.py b/modules/api/nvml.py index 3554e62c8..ba9d9b12f 100644 --- a/modules/api/nvml.py +++ b/modules/api/nvml.py @@ -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)