diff --git a/engine.py b/engine.py index 55b457b..766ed87 100644 --- a/engine.py +++ b/engine.py @@ -116,8 +116,7 @@ def synthesize( kwargs["cfg_weight"] = cfg_weight with torch.inference_mode(): - with torch.amp.autocast(device_type="cuda", dtype=torch.float16): - wav = chatterbox_model.generate(text=text, **kwargs) + wav = chatterbox_model.generate(text=text, **kwargs) if torch.cuda.is_available(): torch.cuda.synchronize()