diff --git a/scripts/gpu_temperature_protection.py b/scripts/gpu_temperature_protection.py index c6d9d02..74ad790 100644 --- a/scripts/gpu_temperature_protection.py +++ b/scripts/gpu_temperature_protection.py @@ -5,8 +5,15 @@ import subprocess import time import re +if hasattr(shared, "OptionHTML"): # < 1.6.0 support + shared.options_templates.update(shared.options_section(('GPU_temperature_protection', "GPU Temperature"), { + "gpu_temps_sleep_temperature_src_explanation": shared.OptionHTML("""NVIDIA - nvidia-smi is available on Windows and Linux.
+AMD - ROCm-smi is Linux only + """) + })) + shared.options_templates.update(shared.options_section(('GPU_temperature_protection', "GPU Temperature"), { - "gpu_temps_sleep_temperature_src": shared.OptionInfo("NVIDIA - nvidia-smi", "Temperature source mode", gr.Radio, {"choices": ["NVIDIA - nvidia-smi", "AMD GPU - ROCm-smi Linux only"]}), + "gpu_temps_sleep_temperature_src": shared.OptionInfo("NVIDIA - nvidia-smi", "Temperature source mode", gr.Radio, {"choices": ["NVIDIA - nvidia-smi", "AMD - ROCm-smi"]}), "gpu_temps_sleep_enable": shared.OptionInfo(True, "Enable GPU temperature protection"), "gpu_temps_sleep_print": shared.OptionInfo(True, "Print GPU Core temperature while sleeping in terminal"), "gpu_temps_sleep_minimum_interval": shared.OptionInfo(5.0, "GPU temperature monitor minimum interval", gr.Number).info("won't check the temperature again until this amount of seconds have passed"), @@ -63,7 +70,7 @@ class GPUTemperatureProtection(scripts.Script): temperature_src_dict = { "NVIDIA - nvidia-smi": get_gpu_temperature_nvidia_smi, - "AMD GPU - ROCm-smi Linux only": get_gpu_temperature_amd_rocm_smi + "AMD - ROCm-smi": get_gpu_temperature_amd_rocm_smi } @staticmethod