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

improve error message in OpenHardwareMonitor

pull/7/head
w-e-w 2024-05-19 17:58:33 +09:00
parent 61adc45ccd
commit 38de3a9615
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
temperature_sensor_modules/open_hardware_monitor.py
View File

@ -1,6 +1,7 @@
from modules import scripts, shared
from modules import scripts, shared, errors
from pathlib import Path
import urllib.request
import gradio as gr
import zipfile
import launch
import os
@ -46,8 +47,7 @@ def init_open_hardware_monitor():
ohm_computer.Open()
# find the first matching temperature sensor for the specified hardware
if ohm_sensors is None or shared.opts.gpu_temps_sleep_gpu_name not in str(
ohm_hardware.Name):
if ohm_sensors is None or shared.opts.gpu_temps_sleep_gpu_name not in str(ohm_hardware.Name):
for hardware in ohm_computer.Hardware:
if shared.opts.gpu_temps_sleep_gpu_name in str(hardware.Name):
for sensor in hardware.Sensors:
@ -59,11 +59,13 @@ def init_open_hardware_monitor():
# sensor not found
ohm_sensors = None
ohm_hardware = None
print(
f"[Error GPU temperature protection] OpenHardwareMonitor Couldn't find temperature sensor for {shared.opts.gpu_temps_sleep_gpu_name}")
error_message = f"OpenHardwareMonitor Couldn't find temperature sensor for {shared.opts.gpu_temps_sleep_gpu_name}"
gr.Warning(error_message)
print(f"[Error GPU temperature protection] {error_message}")
except Exception as e:
print(f"[Error GPU temperature protection] Failed to initialize OpenHardwareMonitor: {e}")
error_message = f"Failed to initialize OpenHardwareMonitor"
errors.report(f'[Error GPU temperature protection] {error_message}')
def get_gpu_temperature_open_hardware_monitor():