Skip to content

Commit 8c9ae64

Browse files
authored
1 parent ede896e commit 8c9ae64

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

DeepFilterNet/df/checkpoint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def read_cp(
7474
epoch = get_epoch(latest)
7575
if log:
7676
logger.info("Found checkpoint {} with epoch {}".format(latest, epoch))
77-
latest = torch.load(latest, map_location="cpu")
77+
latest = torch.load(latest, map_location="cpu", weights_only=True)
7878
latest = {k.replace("clc", "df"): v for k, v in latest.items()}
7979
if blacklist:
8080
reg = re.compile("".join(f"({b})|" for b in blacklist)[:-1])

0 commit comments

Comments
 (0)