jhj0517
commited on
Commit
·
71b7d31
1
Parent(s):
22b4354
Add path
Browse files- modules/utils/paths.py +3 -1
modules/utils/paths.py
CHANGED
|
@@ -4,12 +4,14 @@ import os
|
|
| 4 |
|
| 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 |
|
| 8 |
|
| 9 |
@functools.lru_cache
|
| 10 |
def init_dirs():
|
| 11 |
for dir_path in [
|
| 12 |
-
MODELS_DIR
|
|
|
|
| 13 |
]:
|
| 14 |
os.makedirs(dir_path, exist_ok=True)
|
| 15 |
|
|
|
|
| 4 |
|
| 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
|
| 11 |
def init_dirs():
|
| 12 |
for dir_path in [
|
| 13 |
+
MODELS_DIR,
|
| 14 |
+
OUTPUTS_DIR
|
| 15 |
]:
|
| 16 |
os.makedirs(dir_path, exist_ok=True)
|
| 17 |
|