extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-22 00:03:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow any version of torch/torchvision

Browse Source
This commit is contained in:
cmdr2 2023-04-20 18:40:45 +05:30
parent 023b78d1c9
commit 526fc989c1
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/check_modules.py
View File

@ -59,7 +59,15 @@ def init():
continue
allowed_versions, latest_version = get_allowed_versions(module_name, allowed_versions)
if version(module_name) not in allowed_versions:
requires_install = False
if module_name in ("torch", "torchvision"):
if version(module_name) is None: # allow any torch version
requires_install = True
elif version(module_name) not in allowed_versions:
requires_install = True
if requires_install:
try:
install(module_name, latest_version)
except: