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-06-07 21:56:14 -05:00
parent 7d6fff24f9
commit e35a91c67a
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
vall_e/config.py
View File

@ -99,9 +99,8 @@ class BaseConfig:
state = {} state = {}
if args.yaml: if args.yaml:
cfg_path = args.yaml cfg_path = args.yaml
state = yaml.safe_load(open(cfg_path, "r", encoding="utf-8")) state = yaml.safe_load(open(cfg_path, "r", encoding="utf-8"))
state.setdefault("cfg_path", cfg_path) state.setdefault("cfg_path", cfg_path.parent)
return cls(**state) return cls(**state)
@ -670,9 +669,9 @@ class Config(BaseConfig):
if self.distributed: if self.distributed:
self.dataset.hdf5_flag = "r" self.dataset.hdf5_flag = "r"
try: try:
self.hdf5 = h5py.File(f'{self.cfg_path}/{self.dataset.hdf5_name}', 'a' if write else self.dataset.hdf5_flag) # to-do, have an easy to set flag that determines if training or creating the dataset self.hdf5 = h5py.File(f'{self.relpath}/{self.dataset.hdf5_name}', 'a' if write else self.dataset.hdf5_flag) # to-do, have an easy to set flag that determines if training or creating the dataset
except Exception as e: except Exception as e:
print("Error while opening HDF5 file:", f'{self.cfg_path}/{self.dataset.hdf5_name}', str(e)) print("Error while opening HDF5 file:", f'{self.relpath}/{self.dataset.hdf5_name}', str(e))
self.dataset.use_hdf5 = False self.dataset.use_hdf5 = False
def format( self ): def format( self ):