rahul7star commited on
Commit
ce3d0b2
·
verified ·
1 Parent(s): 9e7b9d5

Update src/chatterbox/mtl_tts.py

Browse files
Files changed (1) hide show
  1. src/chatterbox/mtl_tts.py +7 -7
src/chatterbox/mtl_tts.py CHANGED
@@ -232,16 +232,16 @@ class ChatterboxMultilingualTTS:
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
 
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