extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-11 20:36:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

Tell the user to reinstall ED if they're trying to use a 50xx GPU with Python 3.8-based Easy Diffusion

Browse Source
This commit is contained in:
cmdr2 2025-04-27 21:28:56 +05:30
parent adfe24fdd0
commit 40801de615
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
scripts/check_modules.py
View File

@ -74,8 +74,17 @@ def update_modules():
gpu_infos = get_gpus()
device_names = set(gpu.device_name for gpu in gpu_infos)
if any(BLACKWELL_DEVICES.search(device_name) for device_name in device_names):
print("Upgrading torch to support NVIDIA 50xx series of graphics cards")
torchruntime.install(["--force", "--upgrade", "torch", "torchvision"])
if sys.version_info < (3, 9):
print(
"\n###################################\n"
"NVIDIA 50xx series of graphics cards detected!\n\n"
"To use this graphics card, please install the latest version of Easy Diffusion from: https://github.com/easydiffusion/easydiffusion#installation"
"\n###################################\n"
)
sys.exit()
else:
print("Upgrading torch to support NVIDIA 50xx series of graphics cards")
torchruntime.install(["--force", "--upgrade", "torch", "torchvision"])
for module_name, allowed_versions in modules_to_check.items():
if os.path.exists(f"src/{module_name}"):