fix(chatterbox): disable MIOpen backend to avoid broken RNN kernel
Esse commit está contido em:
@@ -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"))
|
||||
|
||||
|
||||
Referência em uma Nova Issue
Bloquear um usuário