from local

This commit is contained in:
dohe0342 2023-05-24 11:52:43 +09:00
parent aeb474bfe1
commit 1d60614f12
2 changed files with 1 additions and 1 deletions

View File

@ -38,7 +38,7 @@ from convolution import ConvolutionModule
logger = logging.getLogger().setLevel(logging.INFO)
class TransformerEncoderAdapter(TransformerEncoder):
class TransformerEncoderLoRA(TransformerEncoder):
def __init__(self, args: Wav2Vec2Config):
super().__init__(args)
self.lora = LoRAModule()