From 30298b9ca31afe94b8743dfb22faa707eda2ba89 Mon Sep 17 00:00:00 2001
From: mrq <barry.quiggles@protonmail.com>
Date: Thu, 16 Feb 2023 21:36:49 +0000
Subject: [PATCH] fixing brain worms
---
main.py | 9 +++++++++
tortoise/api.py | 8 +-------
2 files changed, 10 insertions(+), 7 deletions(-)
diff --git a/main.py b/main.py
index 8ce0146..cefa8e3 100755
--- a/main.py
+++ b/main.py
@@ -1,5 +1,14 @@
import webui as mrq
+if 'XDG_CACHE_HOME' not in os.environ:
+ os.environ['XDG_CACHE_HOME'] = os.path.realpath('./models/')
+
+if 'TORTOISE_MODELS_DIR' not in os.environ:
+ os.environ['TORTOISE_MODELS_DIR'] = os.path.realpath('./models/tortoise/')
+
+if 'TRANSFORMERS_CACHE' not in os.environ:
+ os.environ['TRANSFORMERS_CACHE'] = os.path.realpath('./models/transformers/')
+
if __name__ == "__main__":
mrq.args = mrq.setup_args()
diff --git a/tortoise/api.py b/tortoise/api.py
index 7ca0a74..cc006a9 100755
--- a/tortoise/api.py
+++ b/tortoise/api.py
@@ -6,12 +6,6 @@ import gc
from time import time
from urllib import request
-if 'TORTOISE_MODELS_DIR' not in os.environ:
- os.environ['TORTOISE_MODELS_DIR'] = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../models/tortoise/')
-
-if 'TRANSFORMERS_CACHE' not in os.environ:
- os.environ['TRANSFORMERS_CACHE'] = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../models/transformers/')
-
import torch
import torch.nn.functional as F
import progressbar
@@ -38,7 +32,7 @@ pbar = None
STOP_SIGNAL = False
-MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR')
+MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR', os.path.realpath('./models/tortoise/'))
MODELS = {
'autoregressive.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/autoregressive.pth',
'classifier.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/classifier.pth',