Nighthawk/soprano-factory
1
0
Fork 0
mirror of https://github.com/Nighthawk42/soprano-factory.git synced 2026-08-30 04:30:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow loading encoder with strict=False

Browse Source
This commit is contained in:
Nighthawk
2026-01-18 02:07:30 -05:00
committed by GitHub
parent 9b7dd275c7
commit 5fbe02a496
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generate_dataset.py
+1 -1
View File
@@ -44,7 +44,7 @@ def main():
print(f"Loading model onto {device}.")
encoder = Encoder().to(device)
encoder_path = hf_hub_download(repo_id='ekwek/Soprano-Encoder', filename='encoder.pth')
encoder.load_state_dict(torch.load(encoder_path, map_location=device))
encoder.load_state_dict(torch.load(encoder_path, map_location=device), strict=False)
encoder.eval()
print("Model loaded.")