mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-03 00:33:57 +03:00
16 lines
318 B
Python
16 lines
318 B
Python
from typing import Protocol
|
|
|
|
import torch
|
|
from torch import device
|
|
|
|
|
|
class EmbedderProtocol(Protocol):
|
|
|
|
def loadModel(self, file: str, dev: device, isHalf: bool = True):
|
|
...
|
|
|
|
def extractFeatures(
|
|
self, feats: torch.Tensor, embOutputLayer=9, useFinalProj=True
|
|
) -> torch.Tensor:
|
|
...
|