Spaces:
Running
Running
reducing model downloads
Browse files- tortoise/models/utils.py +2 -3
tortoise/models/utils.py
CHANGED
@@ -11,9 +11,8 @@ import progressbar
|
|
11 |
|
12 |
D_STEM = "https://drive.google.com/uc?id="
|
13 |
|
14 |
-
DEFAULT_MODELS_DIR =
|
15 |
-
|
16 |
-
)
|
17 |
MODELS_DIR = os.environ.get("TORTOISE_MODELS_DIR", DEFAULT_MODELS_DIR)
|
18 |
|
19 |
# MODELS_DIR = os.environ.get("TORTOISE_MODELS_DIR")
|
|
|
11 |
|
12 |
D_STEM = "https://drive.google.com/uc?id="
|
13 |
|
14 |
+
DEFAULT_MODELS_DIR = "/app/tortoise/models/pretrained_models"
|
15 |
+
|
|
|
16 |
MODELS_DIR = os.environ.get("TORTOISE_MODELS_DIR", DEFAULT_MODELS_DIR)
|
17 |
|
18 |
# MODELS_DIR = os.environ.get("TORTOISE_MODELS_DIR")
|