Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add cache

This commit is contained in:
shahules786 2022-11-10 16:01:06 +05:30
parent 252d380acc
commit a2e083b315
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
enhancer/models/model.py
View File

@ -18,8 +18,11 @@ from enhancer.inference import Inference
from enhancer.loss import LOSS_MAP, LossWrapper
from enhancer.version import __version__
CACHE_DIR = ""
HF_TORCH_WEIGHTS = ""
CACHE_DIR = os.getenv(
"ENHANCER_CACHE",
os.path.expanduser("~/.cache/torch/enhancer"),
)
HF_TORCH_WEIGHTS = "pytorch_model.ckpt"
DEFAULT_DEVICE = "cpu"