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

do not utilize diskcache if a config yaml is not loaded

Browse Source
This commit is contained in:
mrq 2023-08-23 11:02:15 -05:00
parent 524d289c9c
commit d106598403
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vall_e/config.py
View File

@ -428,7 +428,6 @@ class BitsAndBytes:
@dataclass() @dataclass()
class Config(_Config): class Config(_Config):
device: str = "cuda" device: str = "cuda"
#distributed: bool = False
dataset: Dataset = field(default_factory=lambda: Dataset) dataset: Dataset = field(default_factory=lambda: Dataset)
models: Models = field(default_factory=lambda: Models) models: Models = field(default_factory=lambda: Models)
@ -456,7 +455,7 @@ class Config(_Config):
@cached_property @cached_property
def diskcache(self): def diskcache(self):
if self.dataset.cache: if self.cfg_path is not None and self.dataset.cache:
return diskcache.Cache(self.cache_dir).memoize return diskcache.Cache(self.cache_dir).memoize
return lambda: lambda x: x return lambda: lambda x: x