zhzluke96
commited on
Commit
•
2be0618
1
Parent(s):
7f02ca0
update
Browse files
modules/speaker.py
CHANGED
@@ -25,7 +25,7 @@ class Speaker:
|
|
25 |
|
26 |
@staticmethod
|
27 |
def from_tensor(tensor):
|
28 |
-
speaker = Speaker(
|
29 |
speaker.emb = tensor
|
30 |
return speaker
|
31 |
|
@@ -124,7 +124,9 @@ class SpeakerManager:
|
|
124 |
):
|
125 |
if filename == "":
|
126 |
filename = name
|
127 |
-
speaker = Speaker(
|
|
|
|
|
128 |
if isinstance(tensor, torch.Tensor):
|
129 |
speaker.emb = tensor
|
130 |
if isinstance(tensor, list):
|
|
|
25 |
|
26 |
@staticmethod
|
27 |
def from_tensor(tensor):
|
28 |
+
speaker = Speaker(seed_or_tensor=-2)
|
29 |
speaker.emb = tensor
|
30 |
return speaker
|
31 |
|
|
|
124 |
):
|
125 |
if filename == "":
|
126 |
filename = name
|
127 |
+
speaker = Speaker(
|
128 |
+
seed_or_tensor=-2, name=name, gender=gender, describe=describe
|
129 |
+
)
|
130 |
if isinstance(tensor, torch.Tensor):
|
131 |
speaker.emb = tensor
|
132 |
if isinstance(tensor, list):
|
modules/webui/speaker/speaker_creator.py
CHANGED
@@ -64,7 +64,7 @@ def create_spk_from_seed(
|
|
64 |
chat_tts = load_chat_tts()
|
65 |
with SeedContext(seed, True):
|
66 |
emb = chat_tts.sample_random_speaker()
|
67 |
-
spk = Speaker(
|
68 |
spk.emb = emb
|
69 |
|
70 |
with tempfile.NamedTemporaryFile(delete=False, suffix=".pt") as tmp_file:
|
|
|
64 |
chat_tts = load_chat_tts()
|
65 |
with SeedContext(seed, True):
|
66 |
emb = chat_tts.sample_random_speaker()
|
67 |
+
spk = Speaker(seed_or_tensor=-2, name=name, gender=gender, describe=desc)
|
68 |
spk.emb = emb
|
69 |
|
70 |
with tempfile.NamedTemporaryFile(delete=False, suffix=".pt") as tmp_file:
|