Spaces:
Runtime error
Runtime error
File size: 1,962 Bytes
8fd2f2f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from typing import Any, Union, List
class ModuleLoaderConfig:
def __init__(self,
loader_cls_path: str,
cls_func: str = "",
cls_func_fast_dev_run: str = "",
kwargs_diffusers: dict[str, Any] = None,
# model kwargs. Can be just a dict, or a parameter class (derived from modules.params.params_mixin.AsDictMixin) so we have verification of inputs
model_params: Any = None,
# kwargs activated only if on fast_dev_run mode
model_params_fast_dev_run: Any = None,
# load parameters specified in diff_trainer_params (so it links them)
kwargs_diff_trainer_params: dict[str,
Union[str, None]] = None,
args: List[Any] = [],
# names of dependent modules that we need as input
dependent_modules: dict[str, str] = None,
# names of dependent modules that we need as input. Modules will be cloned
dependent_modules_cloned: dict[str, str] = None,
state_dict_path: str = "",
strict_loading: bool = True,
state_dict_filters: List[str] = []
) -> None:
self.loader_cls_path = loader_cls_path
self.cls_func = cls_func
self.cls_func_fast_dev_run = cls_func_fast_dev_run
self.kwargs_diffusers = kwargs_diffusers
self.dependent_modules = dependent_modules
self.dependent_modules_cloned = dependent_modules_cloned
self.kwargs_diff_trainer_params = kwargs_diff_trainer_params
self.model_params = model_params
self.state_dict_path = state_dict_path
self.strict_loading = strict_loading
self.state_dict_filters = state_dict_filters
self.model_params_fast_dev_run = model_params_fast_dev_run
self.args = args
|