CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-07-28 16:08:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Restore map_location = 'cpu'

Browse Source
This commit is contained in:
henryruhs
2025-03-11 14:43:09 +01:00
parent ac41bab3a2
commit a6e1405c70
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/training.py
+2 -2
View File
@@ -26,8 +26,8 @@ class EmbeddingConverterTrainer(lightning.LightningModule):
target_path = CONFIG.get('training.model', 'target_path')
self.embedding_converter = EmbeddingConverter()
self.source_embedder = torch.jit.load(source_path) # type:ignore[no-untyped-call]
self.target_embedder = torch.jit.load(target_path) # type:ignore[no-untyped-call]
self.source_embedder = torch.jit.load(source_path, map_location = 'cpu') # type:ignore[no-untyped-call]
self.target_embedder = torch.jit.load(target_path, map_location = 'cpu') # type:ignore[no-untyped-call]
self.source_embedder.eval()
self.target_embedder.eval()
self.mse_loss = nn.MSELoss()