1580158
/
stable-diffusion-webui-GPU-temperature-protection
mirror of https://github.com/w-e-w/stable-diffusion-webui-GPU-temperature-protection.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add onChangeMethod and move downlode method from install.py to gpu_temperature_protection.py

pull/5/head
ibrahem 2023-09-11 23:35:34 +03:00
parent 33c19aabcd
commit 6b6891c11a
2 changed files with 29 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/gpu_temperature_protection.py
View File

@ -4,10 +4,13 @@ import subprocess
import time
import re
import os
from scripts import constant
from scripts import constant,settings_storage,install
import launch
if shared.opts.gpu_temps_sleep_temperature_src == 'NVIDIA & AMD - openHardwareMonitor' and os.name == 'nt' and launch.is_installed("pythonnet"):
# check is OpenHardwareMonitorLib exist if not will download
if not os.path.isfile(constant.OpenHardwareMonitorLibdllFilePath):
install.downloadOpenHardwareMonitorLib()
import clr # the pythonnet module.
clr.AddReference(constant.openHardwareMonitorLibPath)
from OpenHardwareMonitor.Hardware import Computer
@ -90,6 +93,10 @@ class GPUTemperatureProtection(scripts.Script):
print(f'\n[Error GPU temperature protection] openHardwareMonitor : {e}')
return 0
@staticmethod
def onChangeGpuTempsSleepTemperatureSrc():
settings_storage.settingsStorage.set("gpu_temps_sleep_temperature_src",shared.opts.gpu_temps_sleep_temperature_src)
settings_storage.settingsStorage.save()
temperature_src_dict = {
"NVIDIA - nvidia-smi": get_gpu_temperature_nvidia_smi,
@ -154,7 +161,7 @@ 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": shared.OptionInfo("NVIDIA - nvidia-smi", "Temperature source", gr.Radio, {"choices": list(GPUTemperatureProtection.temperature_src_dict.keys())}),
"gpu_temps_sleep_temperature_src": shared.OptionInfo("NVIDIA - nvidia-smi", "Temperature source", gr.Radio, {"choices": list(GPUTemperatureProtection.temperature_src_dict.keys())}, onchange=GPUTemperatureProtection.onChangeGpuTempsSleepTemperatureSrc).needs_restart(),
"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"),
@ -169,5 +176,5 @@ if os.name == 'nt':
all_lines = subprocess.check_output(['cmd.exe', '/c', 'wmic path win32_VideoController get name']).decode().strip("\nName").splitlines()
names_list = [name.rstrip() for name in all_lines if not re.compile("^ +$").match(name) and name != '']
shared.options_templates.update(shared.options_section(('GPU_temperature_protection', "GPU Temperature"), {
"gpu_temps_sleep_gpu_name": shared.OptionInfo( "none" if len(names_list) == 0 else names_list[0] , "GPU Name", gr.Radio, {"choices": names_list, "interactive":shared.opts.gpu_temps_sleep_temperature_src == 'NVIDIA & AMD - openHardwareMonitor' }).info("select your gpu, only for openHardwareMonitor"),
"gpu_temps_sleep_gpu_name": shared.OptionInfo( "none" if len(names_list) == 0 else names_list[0] , "GPU Name", gr.Radio, {"choices": names_list, "interactive":shared.opts.gpu_temps_sleep_temperature_src == 'NVIDIA & AMD - openHardwareMonitor' }).info("select your gpu, <b>only for openHardwareMonitor - windows</b>"),
}))

34
scripts/install.py
View File

@ -1,22 +1,26 @@
import launch
import os
from modules import shared
from pathlib import Path
from scripts import constant
from scripts import constant,settings_storage
import urllib
import zipfile
# install pythonnet required for openHardwareMonitor Lib
if shared.opts.gpu_temps_sleep_temperature_src == 'NVIDIA & AMD - openHardwareMonitor' and os.name == 'nt':
if not launch.is_installed("pythonnet"):
launch.run_pip("install pythonnet==3.0.2", "requirements for windows OpenHardwareMonitorLib")
# check and create OpenHardwareMonitor folder
Path(constant.openHardwareMonitorDirPath).mkdir(parents=True, exist_ok=True)
# check is OpenHardwareMonitorLib exist if not will download
if not os.path.isfile(constant.OpenHardwareMonitorLibdllFilePath):
zip_path, _ = urllib.request.urlretrieve(constant.OpenHardwareMonitorLibDownloadUrl)
with zipfile.ZipFile(zip_path, "r") as z:
with open(os.path.realpath(constant.OpenHardwareMonitorLibdllFilePath) , 'wb') as f:
f.write(z.read('OpenHardwareMonitor/OpenHardwareMonitorLib.dll'))
def downloadOpenHardwareMonitorLib():
zip_path, _ = urllib.request.urlretrieve(constant.OpenHardwareMonitorLibDownloadUrl)
with zipfile.ZipFile(zip_path, "r") as z:
with open(os.path.realpath(constant.OpenHardwareMonitorLibdllFilePath) , 'wb') as f:
f.write(z.read('OpenHardwareMonitor/OpenHardwareMonitorLib.dll'))
# install pythonnet required for openHardwareMonitor Lib
# shared.opts.gpu_temps_sleep_temperature_src == 'NVIDIA & AMD - openHardwareMonitor' and
if os.name == 'nt' :
if settings_storage.settingsStorage.get("gpu_temps_sleep_temperature_src") == "NVIDIA & AMD - openHardwareMonitor":
if not launch.is_installed("pythonnet"):
launch.run_pip("install pythonnet==3.0.2", "requirements for windows OpenHardwareMonitorLib")
# check and create OpenHardwareMonitor folder
Path(constant.openHardwareMonitorDirPath).mkdir(parents=True, exist_ok=True)