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 torch 1.11 to continue being installed

Browse Source
This commit is contained in:
cmdr2 2023-04-18 16:10:46 +05:30
parent d8d44c579c
commit 30a133bad9
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/check_modules.py
View File

@ -1,5 +1,9 @@
"""
This script checks and installs the required modules.
TODO - Maybe replace the bulk of this script with a call to `pip install -f requirements.txt`, with
a custom index URL depending on the platform.
"""
import os
@ -9,8 +13,8 @@ import platform
os_name = platform.system()
modules_to_check = {
"torch": ("1.13.1", "2.0.0"),
"torchvision": ("0.14.1", "0.15.1"),
"torch": ("1.11.0", "1.13.1", "2.0.0"),
"torchvision": ("0.12.0", "0.14.1", "0.15.1"),
"sdkit": "1.0.72",
"stable-diffusion-sdkit": "2.1.4",
"rich": "12.6.0",