| | |
|---|
| | | self.spk_model = spk_model |
|---|
| | | self.spk_kwargs = spk_kwargs |
|---|
| | | self.model_path = kwargs.get("model_path") |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | def build_model(self, **kwargs): |
|---|
| | | assert "model" in kwargs |
|---|
| | |
|---|
| | | set_all_random_seed(kwargs.get("seed", 0)) |
|---|
| | | |
|---|
| | | device = kwargs.get("device", "cuda") |
|---|
| | | if not torch.cuda.is_available() or kwargs.get("ngpu", 0) == 0: |
|---|
| | | if not torch.cuda.is_available() or kwargs.get("ngpu", 1) == 0: |
|---|
| | | device = "cpu" |
|---|
| | | kwargs["batch_size"] = 1 |
|---|
| | | kwargs["device"] = device |
|---|
| | |
|---|
| | | kwargs.update(cfg) |
|---|
| | | res = self.model(*args, kwargs) |
|---|
| | | return res |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | def generate(self, input, input_len=None, **cfg): |
|---|
| | | if self.vad_model is None: |
|---|
