From 8bd5eb5ce4f1fbc2d53db9efd823ad5124571eb4 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Wed, 9 Oct 2024 18:19:21 +0530 Subject: [PATCH] Remove unnecessary patch for sdkit's model downloader. ED now handles both kinds of model folder names --- ui/easydiffusion/model_manager.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/ui/easydiffusion/model_manager.py b/ui/easydiffusion/model_manager.py index 2434f5e0..6e8c9325 100644 --- a/ui/easydiffusion/model_manager.py +++ b/ui/easydiffusion/model_manager.py @@ -479,17 +479,3 @@ def get_model_dirs(model_type: str, base_dir=None): dirs.append(alt_dir) 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