mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
16 lines
394 B
Python
16 lines
394 B
Python
from typing import Any, Dict, List, Tuple
|
|
|
|
from numpy.typing import NDArray
|
|
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]]
|
|
LossDict = Dict[str, Tensor]
|
|
IDEmbedding = Tensor
|
|
|
|
Embedding = NDArray[Any]
|
|
VisionFrame = NDArray[Any]
|