Update app.py
Browse files
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
|