Alstears commited on
Commit
0b8917a
·
verified ·
1 Parent(s): a6ba84e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -23,11 +23,10 @@ def get_model():
23
  if _model is None:
24
  with _lock:
25
  if _model is None:
26
- m = ChatterboxTTS.from_pretrained(device="cpu")
27
  ckpt = hf_hub_download(repo_id=MODEL_REPO, filename=CHECKPOINT_FILENAME)
28
  t3_state = load_file(ckpt, device="cpu")
29
  m.t3.load_state_dict(t3_state)
30
- m = m.to("cpu")
31
  m.eval()
32
  _model = m
33
  return _model
 
23
  if _model is None:
24
  with _lock:
25
  if _model is None:
26
+ m = ChatterboxTTS.from_pretrained(device="cpu") # device di sini saja
27
  ckpt = hf_hub_download(repo_id=MODEL_REPO, filename=CHECKPOINT_FILENAME)
28
  t3_state = load_file(ckpt, device="cpu")
29
  m.t3.load_state_dict(t3_state)
 
30
  m.eval()
31
  _model = m
32
  return _model