add limitation

lxning · lxning · commit caa5042135c6 · 2024-02-17T21:49:49.000-08:00
diff --git a/examples/cpp/aot_inductor/bert/aot_compile_export.py b/examples/cpp/aot_inductor/bert/aot_compile_export.py
@@ -11,8 +11,9 @@
 )
 
 set_seed(1)
+# PT2.2 has limitation on the max
 MAX_BATCH_SIZE = 15
-MAX_LENGTH = 1024
+MAX_LENGTH = 511
 
 
 def transformers_model_dowloader(

Original file line number	Diff line number	Diff line change
`@@ -11,8 +11,9 @@`
`11`	`11`	`)`
`12`	`12`
`13`	`13`	`set_seed(1)`
	`14`	`+# PT2.2 has limitation on the max`
`14`	`15`	`MAX_BATCH_SIZE = 15`
`15`		`-MAX_LENGTH = 1024`
	`16`	`+MAX_LENGTH = 511`
`16`	`17`
`17`	`18`
`18`	`19`	`def transformers_model_dowloader(`