type hints

2023-07-10 20:48:32 +03:00 · 2023-07-10 20:48:32 +03:00 · 341ef458b4
commit 341ef458b4
parent a2063b7000
1 changed files with 1 additions and 1 deletions
--- a/torchscale/architecture/encoder_decoder.py
+++ b/torchscale/architecture/encoder_decoder.py
@ -11,7 +11,7 @@ from torchscale.architecture.encoder import Encoder
 class EncoderDecoder(nn.Module):
    def __init__(
        self,
-        args,
+        args: EncoderDecoderConfig,
        encoder_embed_tokens=None,
        encoder_embed_positions=None,
        decoder_embed_tokens=None,