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

relax type checks

Browse Source
This commit is contained in:
Vladimir Mandic
2023-11-23 14:23:19 -05:00
parent 88aabcb3c5
commit aed6aa0217
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/merging/merge.py
+2 -2
View File
@@ -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,