extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-08-08 23:44:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove hardcoded references to torch.cuda; Use torchruntime and sdkit's device utilities instead

Browse Source
This commit is contained in:
cmdr2
2025-02-06 13:06:51 +05:30
parent 2e4623736a
commit cd8365558a
7 changed files with 83 additions and 177 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/check_modules.py
View File

@ -73,6 +73,7 @@ modules_to_check = {
# "xformers": "0.0.16",
"huggingface-hub": "0.21.4",
"wandb": "0.13.7",
"torchruntime": "1.8.0",
}
modules_to_log = ["torch", "torchvision", "sdkit", "stable-diffusion-sdkit", "diffusers"]
@ -168,10 +169,10 @@ def update_modules():
# if sdkit is 2.0.15.x (or lower), then diffusers should be restricted to 0.21.4 (see below for the reason)
# otherwise use the current sdkit version (with the corresponding diffusers version)
expected_sdkit_version_str = "2.0.22.3"
expected_sdkit_version_str = "2.0.22.4"
expected_diffusers_version_str = "0.28.2"
legacy_sdkit_version_str = "2.0.15.12"
legacy_sdkit_version_str = "2.0.15.13"
legacy_diffusers_version_str = "0.21.4"
sdkit_version_str = version("sdkit")