Update src/chatterbox/mtl_tts.py
Browse files
src/chatterbox/mtl_tts.py
CHANGED
|
@@ -232,16 +232,16 @@ class ChatterboxMultilingualTTS:
|
|
| 232 |
],
|
| 233 |
token=os.getenv("HF_TOKEN"),
|
| 234 |
)
|
| 235 |
-
|
| 236 |
|
| 237 |
-
|
| 238 |
|
| 239 |
-
|
| 240 |
-
|
| 241 |
model = model.to("cpu")
|
| 242 |
|
| 243 |
-
|
| 244 |
for p in model.parameters():
|
| 245 |
-
|
| 246 |
|
| 247 |
-
|
|
|
|
| 232 |
],
|
| 233 |
token=os.getenv("HF_TOKEN"),
|
| 234 |
)
|
| 235 |
+
)
|
| 236 |
|
| 237 |
+
model = cls.from_local(ckpt_dir, device)
|
| 238 |
|
| 239 |
+
# Extra safety: force model tensors to CPU
|
| 240 |
+
if hasattr(model, "to"):
|
| 241 |
model = model.to("cpu")
|
| 242 |
|
| 243 |
+
model.eval()
|
| 244 |
for p in model.parameters():
|
| 245 |
+
p.requires_grad = False
|
| 246 |
|
| 247 |
+
return model
|