We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ede896e commit 8c9ae64Copy full SHA for 8c9ae64
DeepFilterNet/df/checkpoint.py
@@ -74,7 +74,7 @@ def read_cp(
74
epoch = get_epoch(latest)
75
if log:
76
logger.info("Found checkpoint {} with epoch {}".format(latest, epoch))
77
- latest = torch.load(latest, map_location="cpu")
+ latest = torch.load(latest, map_location="cpu", weights_only=True)
78
latest = {k.replace("clc", "df"): v for k, v in latest.items()}
79
if blacklist:
80
reg = re.compile("".join(f"({b})|" for b in blacklist)[:-1])
0 commit comments