mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
Modernize to use ModuleList, Fix some types
This commit is contained in:
@@ -1,13 +1,13 @@
|
||||
from typing import Any, Tuple
|
||||
from typing import Any, Tuple, TypeAlias
|
||||
|
||||
from numpy.typing import NDArray
|
||||
from torch import Tensor
|
||||
|
||||
Embedding = NDArray[Any]
|
||||
EmbeddingDataset = NDArray[Embedding]
|
||||
FaceLandmark5 = NDArray[Any]
|
||||
Embedding : TypeAlias = NDArray[Any]
|
||||
EmbeddingDataset : TypeAlias = NDArray[Embedding]
|
||||
FaceLandmark5 : TypeAlias = NDArray[Any]
|
||||
|
||||
VisionFrame = NDArray[Any]
|
||||
VisionTensor = Tensor
|
||||
VisionFrame : TypeAlias = NDArray[Any]
|
||||
VisionTensor : TypeAlias = Tensor
|
||||
|
||||
Batch = Tuple[VisionTensor, VisionTensor]
|
||||
Batch : TypeAlias = Tuple[VisionTensor, VisionTensor]
|
||||
|
||||
Reference in New Issue
Block a user