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 []