mrq/torchscale
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

type hints

Browse Source
This commit is contained in:
Alexander Goryunov 2023-07-10 20:48:32 +03:00
parent a2063b7000
commit 341ef458b4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchscale/architecture/encoder_decoder.py
View File

@ -11,7 +11,7 @@ from torchscale.architecture.encoder import Encoder
class EncoderDecoder(nn.Module): class EncoderDecoder(nn.Module):
def __init__( def __init__(
self, self,
args, args: EncoderDecoderConfig,
encoder_embed_tokens=None, encoder_embed_tokens=None,
encoder_embed_positions=None, encoder_embed_positions=None,
decoder_embed_tokens=None, decoder_embed_tokens=None,