From e9c87060df4f30278315289dab54e2be31732572 Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 31 May 2024 22:22:28 -0500 Subject: [PATCH] oops --- vall_e/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/config.py b/vall_e/config.py index 8df7b57..4adf307 100755 --- a/vall_e/config.py +++ b/vall_e/config.py @@ -214,7 +214,7 @@ class Model: audio_embedding_sums: bool = True dropout: float = 0.1 # adjustable dropout value loss_factors: dict = field(default_factory=lambda: { "text": 0.1, "prom": 0.0, "resp": 1.0 }) - kv_heads: int = 4 + kv_heads: int = 0 def get(self, name=None): return [ self ] if not name or self.name == name else []