mirror of
https://github.com/easydiffusion/easydiffusion.git
synced 2024-11-26 02:05:09 +01:00
Remove unnecessary patch for sdkit's model downloader. ED now handles both kinds of model folder names
This commit is contained in:
parent
78dcc7cb03
commit
8bd5eb5ce4
@ -479,17 +479,3 @@ def get_model_dirs(model_type: str, base_dir=None):
|
|||||||
dirs.append(alt_dir)
|
dirs.append(alt_dir)
|
||||||
|
|
||||||
return dirs
|
return dirs
|
||||||
|
|
||||||
|
|
||||||
# patch sdkit
|
|
||||||
def __patched__get_actual_base_dir(model_type, download_base_dir, subdir_for_model_type):
|
|
||||||
"Patched version that works with case-insensitive model sub-dirs"
|
|
||||||
|
|
||||||
download_base_dir = os.path.join("~", ".cache", "sdkit") if download_base_dir is None else download_base_dir
|
|
||||||
download_base_dir = get_model_dirs(model_type, download_base_dir)[0] if subdir_for_model_type else download_base_dir
|
|
||||||
return os.path.abspath(download_base_dir)
|
|
||||||
|
|
||||||
|
|
||||||
from sdkit.models import model_downloader
|
|
||||||
|
|
||||||
model_downloader.get_actual_base_dir = __patched__get_actual_base_dir
|
|
||||||
|
Loading…
Reference in New Issue
Block a user