r3gm commited on
Commit
f39cb66
1 Parent(s): 150f0a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()