diff --git a/modules/merging/merge.py b/modules/merging/merge.py index 68d4b2c21..7452cb3e6 100644 --- a/modules/merging/merge.py +++ b/modules/merging/merge.py @@ -72,7 +72,7 @@ def log_vram(txt=""): def load_thetas( - models: Dict[str, os.PathLike | str], + models: Dict[str, os.PathLike], prune: bool, device: torch.device, precision: str, @@ -94,7 +94,7 @@ def load_thetas( def merge_models( - models: Dict[str, os.PathLike | str], + models: Dict[str, os.PathLike], merge_mode: str, precision: str = "fp16", weights_clip: bool = False,