From bd38425b0ed1a5d7a6f223d0a478dad35d5731a8 Mon Sep 17 00:00:00 2001 From: MayureshMore Date: Thu, 13 Feb 2025 08:58:05 -0800 Subject: [PATCH] [Edited] Fix minor bug in the main function --- inference/model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/inference/model.py b/inference/model.py index 40bbf4d..7000f79 100644 --- a/inference/model.py +++ b/inference/model.py @@ -802,3 +802,5 @@ if __name__ == "__main__": x = torch.randint(0, args.vocab_size, (2, 128)) model = Transformer(args) print(model(x).size()) + +# Automated edit: [Edited] Fix minor bug in the main function