Explorar o código

Disable neft on eval

Lengyue %!s(int64=2) %!d(string=hai) anos
pai
achega
415c78a7d4
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      fish_speech/models/text2semantic/llama.py

+ 1 - 1
fish_speech/models/text2semantic/llama.py

@@ -166,7 +166,7 @@ class Transformer(nn.Module):
         x = torch.stack(vocab_embeds, dim=3)
         x = x.sum(dim=3)
 
-        if self.config.neft_alpha > 0:
+        if self.config.neft_alpha > 0 and self.training:
             # alpha / sqrt(L * D)
             scaled_alpha = self.config.neft_alpha / math.sqrt(
                 self.config.dim * x.shape[2]