Update app.py
Browse files
app.py
CHANGED
@@ -1805,11 +1805,11 @@ def __bark__(text, voice_preset):
|
|
1805 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
1806 |
dtype = torch.float32 if "cpu" in device else torch.float16
|
1807 |
bark_processor = AutoProcessor.from_pretrained(
|
1808 |
-
"suno/bark",
|
1809 |
cache_dir=os.path.join(now_dir,"tts","suno/bark"),
|
1810 |
torch_dtype=dtype)
|
1811 |
bark_model = BarkModel.from_pretrained(
|
1812 |
-
"suno/bark",
|
1813 |
cache_dir=os.path.join(now_dir,"tts","suno/bark"),
|
1814 |
torch_dtype=dtype).to(device)
|
1815 |
# bark_model.enable_cpu_offload()
|
|
|
1805 |
device = "cuda:0" if torch.cuda.is_available() else "cpu"
|
1806 |
dtype = torch.float32 if "cpu" in device else torch.float16
|
1807 |
bark_processor = AutoProcessor.from_pretrained(
|
1808 |
+
"suno/bark-small",
|
1809 |
cache_dir=os.path.join(now_dir,"tts","suno/bark"),
|
1810 |
torch_dtype=dtype)
|
1811 |
bark_model = BarkModel.from_pretrained(
|
1812 |
+
"suno/bark-small",
|
1813 |
cache_dir=os.path.join(now_dir,"tts","suno/bark"),
|
1814 |
torch_dtype=dtype).to(device)
|
1815 |
# bark_model.enable_cpu_offload()
|