CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-06-25 07:59:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add config to dataloader alignment

Browse Source
This commit is contained in:
henryruhs
2025-03-03 09:32:44 +01:00
parent 589568bfb5
commit cc9a0ba83e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/dataset.py
+1 -1
View File
@@ -40,7 +40,7 @@ class DynamicDataset(Dataset[Tensor]):
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
])
def warp_tensor(self, temp_tensor : Tensor):
def warp_tensor(self, temp_tensor : Tensor) -> Tensor:
return warp_tensor(temp_tensor.unsqueeze(0), self.warp_matrix).squeeze(0)
def prepare_different_batch(self, source_path : str) -> Batch: