0925 use custom x_transformers for easy develop

2025-09-25 16:04:22 +08:00
parent 6f03357342
commit d077e3210e
6 changed files with 4775 additions and 3 deletions
--- a/Amadeus/transformer_utils.py
+++ b/Amadeus/transformer_utils.py
@ -1,7 +1,7 @@
 import torch
 import torch.nn as nn

-from x_transformers import Decoder, Encoder, PrefixDecoder, CrossAttender
+from Amadeus.custom_x_transformers import Decoder, Encoder, PrefixDecoder, CrossAttender
 from transformers import T5EncoderModel
 from data_representation.vocab_utils import LangTokenVocab