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

that's what I get for testing without hdf5 on my previous machine....

Browse Source
This commit is contained in:
mrq 2024-08-02 20:44:01 -05:00
parent 7a77978096
commit 4456d3172b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vall_e/data.py
View File

@ -808,7 +808,7 @@ class Dataset(_Dataset):
if key not in cfg.hdf5: if key not in cfg.hdf5:
raise RuntimeError(f'Key of Path ({path}) not in HDF5: {key}') raise RuntimeError(f'Key of Path ({path}) not in HDF5: {key}')
metadata = cfg.hdf5[key].attrs() metadata = cfg.hdf5[key].attrs
text = cfg.hdf5[key]["text"][:] text = cfg.hdf5[key]["text"][:]
resps = cfg.hdf5[key]["audio"][:, :] resps = cfg.hdf5[key]["audio"][:, :]
@ -915,7 +915,7 @@ class Dataset(_Dataset):
if key not in cfg.hdf5: if key not in cfg.hdf5:
raise RuntimeError(f'Key of Path ({path}) not in HDF5: {key}') raise RuntimeError(f'Key of Path ({path}) not in HDF5: {key}')
metadata = cfg.hdf5[key].attrs() metadata = cfg.hdf5[key].attrs
text = cfg.hdf5[key]["text"][:] text = cfg.hdf5[key]["text"][:]
resps = cfg.hdf5[key]["audio"][:, :] resps = cfg.hdf5[key]["audio"][:, :]