jhj0517
commited on
Commit
•
47755e8
1
Parent(s):
41e1925
Add model config
Browse files- modules/utils/paths.py +1 -0
modules/utils/paths.py
CHANGED
|
@@ -5,6 +5,7 @@ import os
|
|
| 5 |
PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "..")
|
| 6 |
MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models")
|
| 7 |
OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
@functools.lru_cache
|
|
|
|
| 5 |
PROJECT_ROOT_DIR = os.path.join(os.path.abspath(os.path.dirname(__file__)), "..", "..")
|
| 6 |
MODELS_DIR = os.path.join(PROJECT_ROOT_DIR, "models")
|
| 7 |
OUTPUTS_DIR = os.path.join(PROJECT_ROOT_DIR, "outputs")
|
| 8 |
+
MODEL_CONFIGS = os.path.join(PROJECT_ROOT_DIR, "modules", "config", "models.yaml")
|
| 9 |
|
| 10 |
|
| 11 |
@functools.lru_cache
|