configurable tmp path

This commit is contained in:
Stepan Zuev
2024-04-05 07:42:11 +03:00
parent 6f71fa65fb
commit 21b69ad676
2 changed files with 2 additions and 1 deletions

View File

@@ -15,7 +15,7 @@ import random
import uuid
TMP_PATH = "./demo/temp"
TMP_PATH = os.getenv("TMP_PATH", "./demo/temp")
device = "cuda" if torch.cuda.is_available() else "cpu"
whisper_model, align_model, voicecraft_model = None, None, None