mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
harisreedhar
20 lines
568 B
Python
20 lines
568 B
Python
from collections import OrderedDict
|
|
from typing import Any, Dict, List, Tuple
|
|
|
|
from torch import Tensor
|
|
from torch.utils.data import DataLoader
|
|
|
|
Batch = Tuple[Any, Any, Any]
|
|
Loader = DataLoader[Tuple[Tensor, ...]]
|
|
TargetAttributes = Tuple[Tensor, ...]
|
|
DiscriminatorOutputs = List[List[Tensor]]
|
|
IdEmbedding = Tensor
|
|
SourceEmbedding = IdEmbedding
|
|
StateDict = OrderedDict[str, Any]
|
|
Padding = Tuple[int, int, int, int]
|
|
FaceLandmark203 = Tensor
|
|
VisionTensor = Tensor
|
|
LossTensor = Tensor
|
|
GeneratorLossSet = Dict[str, LossTensor]
|
|
DiscriminatorLossSet = Dict[str, LossTensor]
|