Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update typing for model data access

Browse Source
This commit is contained in:
awsr
2026-03-26 16:42:04 -07:00
parent a6029afb04
commit 775ff59fdb
2 changed files with 19 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared.py
+9 -7
View File
@@ -236,12 +236,14 @@ def restore_defaults(restart=True):
restart_server(restart)
# startup def of shared.sd_model before its redefined in modeldata
sd_model: DiffusionPipeline | None = None # dummy and overwritten by class
sd_refiner: DiffusionPipeline | None = None # dummy and overwritten by class
sd_model_type: str = '' # dummy and overwritten by class
sd_refiner_type: str = '' # dummy and overwritten by class
sd_loaded: bool = False # dummy and overwritten by class
from modules.modeldata import Shared # pylint: disable=ungrouped-imports
from modules.modeldata import Shared # pylint: disable=ungrouped-imports
sys.modules[__name__].__class__ = Shared
if TYPE_CHECKING:
# From Shared class
sd_model: DiffusionPipeline | None
sd_refiner: DiffusionPipeline | None
sd_model_type: str
sd_refiner_type: str
sd_loaded: bool