holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't use rich

Browse Source 
During the first installation, rich is not yet installed
This commit is contained in:
JeLuF 2023-09-13 13:39:34 +02:00
parent 982af1fff3
commit f5e489ba87
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
scripts/check_modules.py
View File

@ -16,9 +16,6 @@ import shutil
from pathlib import Path from pathlib import Path
from pprint import pprint from pprint import pprint
from rich.console import Console
from rich.panel import Panel
os_name = platform.system() os_name = platform.system()
modules_to_check = { modules_to_check = {
@ -217,16 +214,18 @@ def get_config():
def setup_amd_environment(): def setup_amd_environment():
if not os.access("/dev/kfd", os.W_OK): if not os.access("/dev/kfd", os.W_OK):
Console().print( print(
Panel( f"#########################################################################\n"
"\n" + f"# EasyDiffusion has no write access to /dev/kfd. #\n"
+ "[black]EasyDiffusion has no write access to /dev/kfd.\n\n" + f"#########################################################################\n"
+ "Without this, the ROCm driver will probably not be able to initialize the GPU and EasyDiffusion will use the CPU for rendering.\n" + f"# #\n"
+ f"Follow the instructions on this site to configure the access:\n" + f"# Without this, the ROCm driver will probably not be able to initialize #\n"
+ f"[bold yellow underline]https://github.com/easydiffusion/easydiffusion/wiki/AMD-on-Linux#access-permissions\n", + f"# the GPU and EasyDiffusion will use the CPU for rendering. #\n"
title="Missing write access to /dev/kfd", + f"# #\n"
style="white on red", + f"# Follow the instructions on this site to configure the access: #\n"
) + f"# https://github.com/easydiffusion/easydiffusion/wiki/AMD-on-Linux #\n",
+ f"# #\n"
+ f"#########################################################################\n"
) )
gpus = list(filter(lambda x: ("amdgpu" in x), open("/proc/bus/pci/devices", "r").readlines())) gpus = list(filter(lambda x: ("amdgpu" in x), open("/proc/bus/pci/devices", "r").readlines()))