From aed6aa021709b79e6b139ccbde3b969d0d2b3993 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Thu, 23 Nov 2023 14:23:19 -0500 Subject: [PATCH] relax type checks --- modules/merging/merge.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,