mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-02-03 08:43: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:
|
||
|
|
...
|