mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
Henry Ruhs
0722db91f1
* change to convert template * changes * changes * Conditional convert input tensor * Conditional convert input tensor --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
25 lines
724 B
Python
25 lines
724 B
Python
from typing import Any, Dict, Literal, Tuple, TypeAlias
|
|
|
|
from torch import Tensor
|
|
from torch.nn import Module
|
|
|
|
Batch : TypeAlias = Tuple[Tensor, Tensor]
|
|
BatchMode = Literal['equal', 'same', 'different']
|
|
|
|
Feature : TypeAlias = Tensor
|
|
Embedding : TypeAlias = Tensor
|
|
Mask : TypeAlias = Tensor
|
|
Loss : TypeAlias = Tensor
|
|
|
|
Padding : TypeAlias = Tuple[int, int, int, int]
|
|
|
|
GeneratorModule : TypeAlias = Module
|
|
EmbedderModule : TypeAlias = Module
|
|
GazerModule : TypeAlias = Module
|
|
FaceMaskerModule : TypeAlias = Module
|
|
|
|
OptimizerSet : TypeAlias = Any
|
|
|
|
ConvertTemplate = Literal['arcface_128_to_arcface_112_v2', 'ffhq_512_to_arcface_128', 'vggfacehq_512_to_arcface_128']
|
|
ConvertTemplateSet : TypeAlias = Dict[ConvertTemplate, Tensor]
|