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