mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

ugh

Browse Source
This commit is contained in:
mrq 2024-09-18 21:40:57 -05:00
parent fe241f6a99
commit c8d4716a9f
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vall_e/emb/similar.py
View File

@ -195,7 +195,7 @@ def process(
sorted_similarities = {}
for index, filename in tqdm(enumerate(keys), total=len(keys), desc=f"Computing similarities: {speaker_path.name}"):
for index, filename in tqdm(enumerate(keys), total=len(keys), desc=f"Computing similarities: {speaker_path.name}", disable=not verbose):
if features[filename] is None:
continue
@ -241,7 +241,7 @@ def main():
args = parser.parse_args()
args.skip_existing = False #
args.skip_existing = True #
if args.use_dataset:
cfg.metadata_dir.mkdir(parents=True, exist_ok=True)
@ -278,7 +278,7 @@ def main():
dtype=args.dtype,
amp=args.amp,
verbose=True,
verbose=False,
)
if not similarities:

1
vall_e/engines/__init__.py
View File

@ -188,6 +188,7 @@ def load_engines(training=True, **model_kwargs):
keys = [
("text_emb.weight", model.config.text_tokens ),
("tasks_emb.weight", model.config.tasks ),
("langs_emb.weight", model.config.langs ),
("rvq_l_emb.weight", model.config.resp_levels + (1 if "len" in model.config.capabilities else 0) ),
("resps_emb.embeddings.0.weight", model.config.audio_tokens + uses_stop_token ),
("model.embed_tokens.weight", model.config.audio_tokens + uses_stop_token ),