djkesu commited on
Commit
6eb0448
1 Parent(s): 0a3dc8c

reducing model downloads

Browse files
Files changed (1) hide show
  1. 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 = os.path.join(
15
- os.path.expanduser("~"), ".cache", "tortoise", "models"
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")