fix(chatterbox): disable MIOpen backend to avoid broken RNN kernel

Esse commit está contido em:
2026-09-13 00:09:23 +02:00
commit 44565efdaa
+4
Ver Arquivo
@@ -18,6 +18,10 @@ elif torch.cuda.is_available(): # ROCm builds expose the GPU via the CUDA API t
else:
DEVICE = "cpu"
if DEVICE == "cuda":
# avoid MIOpen's broken JIT RNN kernel on some ROCm/GPU combos
torch.backends.cudnn.enabled = False
EXAGGERATION = float(os.environ.get("CHATTERBOX_EXAGGERATION", "0.5"))
CFG_WEIGHT = float(os.environ.get("CHATTERBOX_CFG_WEIGHT", "0.5"))