CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-04-19 15:56:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Simplify Batch type

Browse Source
This commit is contained in:
henryruhs
2025-02-14 16:19:14 +01:00
parent a971506271
commit 650551c19b
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
embedding_converter/src/types.py
+2 -3
View File
@@ -3,11 +3,10 @@ from typing import Any, Tuple, TypeAlias
from numpy.typing import NDArray
from torch import Tensor
Batch : TypeAlias = Tuple[Tensor, Tensor]
Embedding : TypeAlias = NDArray[Any]
EmbeddingDataset : TypeAlias = NDArray[Embedding]
FaceLandmark5 : TypeAlias = NDArray[Any]
VisionFrame : TypeAlias = NDArray[Any]
VisionTensor : TypeAlias = Tensor
Batch : TypeAlias = Tuple[VisionTensor, VisionTensor]
face_swapper/src/types.py
+2 -2
View File
@@ -5,6 +5,8 @@ from numpy.typing import NDArray
from torch import Tensor
from torch.nn import Module
Batch : TypeAlias = Tuple[Tensor, Tensor, Tensor]
ImagePathList : TypeAlias = List[str]
ImagePathSet : TypeAlias = Dict[str, ImagePathList]
@@ -22,8 +24,6 @@ VisionFrame : TypeAlias = NDArray[Any]
LossTensor : TypeAlias = Tensor
VisionTensor : TypeAlias = Tensor
Batch : TypeAlias = Tuple[VisionTensor, VisionTensor, Tensor]
GeneratorLossSet : TypeAlias = Dict[str, Tensor]
DiscriminatorLossSet : TypeAlias = Dict[str, Tensor]