From 253441b75020caf3f59024681a1ed9091dab5f40 Mon Sep 17 00:00:00 2001 From: mrq Date: Sat, 4 May 2024 13:13:52 -0500 Subject: [PATCH] forgot to disable verbose flag --- vall_e/models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/models/base.py b/vall_e/models/base.py index 23d2a71..28f2359 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -197,7 +197,7 @@ try: except Exception as e: print("Error creating `LLamaXformersAttention`:", e) -def replace_attention( model, impl, verbose=True ): +def replace_attention( model, impl, verbose=Valse ): device = next(model.parameters()).device dtype = next(model.parameters()).dtype attentions = [k.split('.') for k, m in model.named_modules() if isinstance(m, LlamaAttention)]