int(max_seq_len) (#5269)

digantdesai · facebook-github-bot · commit 272d4d80fdce · 2024-09-12T11:11:55.000-07:00
Summary: Pull Request resolved: #5269 Reviewed By: larryliu0820 Differential Revision: D62507161 Pulled By: digantdesai fbshipit-source-id: b014a7f87e99323b6d429265a1b1a65d93ba89f8
diff --git a/examples/models/llava/export_llava.py b/examples/models/llava/export_llava.py
@@ -273,6 +273,7 @@ def main():
     parser.add_argument(
         "--max-seq-len",
         default=768,
+        type=int,
         help="Maximum sequence length for the text model.",
     )
     parser.add_argument(

Original file line number	Diff line number	Diff line change
`@@ -273,6 +273,7 @@ def main():`
`273`	`273`	`parser.add_argument(`
`274`	`274`	`"--max-seq-len",`
`275`	`275`	`default=768,`
	`276`	`+ type=int,`
`276`	`277`	`help="Maximum sequence length for the text model.",`
`277`	`278`	`)`
`278`	`279`	`parser.add_argument(`