Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci

Signed-off-by: hsiehjackson <[email protected]>
  • Loading branch information
pre-commit-ci[bot] authored and hsiehjackson committed Apr 14, 2023
1 parent 9e51277 commit ae55bc3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions nemo/collections/tts/modules/submodules.py
Original file line number Diff line number Diff line change
Expand Up @@ -694,14 +694,14 @@ class LookupTable(torch.nn.Module):
"""
LookupTable based Speaker Embedding
"""

def __init__(self, num_speakers, embedding_dim):
super(LookupTable, self).__init__()
self.table = torch.nn.Embedding(num_speakers, embedding_dim)

def forward(self, speaker):
return self.table(speaker)


class SpeakerEncoder(NeuralModule):
"""
Expand All @@ -717,7 +717,7 @@ def __init__(self, lookup_module=None, gst_module=None):
super(SpeakerEncoder, self).__init__()
self.lookup_module = lookup_module
self.gst_module = gst_module

@property
def input_types(self):
return {
Expand Down

0 comments on commit ae55bc3

Please sign in to comment.