enable vit quantization (#417)

kimishpatel · facebook-github-bot · commit 9099f7943abd · 2023-09-20T18:01:30.000-07:00
Summary: Pull Request resolved: #417 After 0919 nightly of pytorch vit linear layers should be quantized Reviewed By: jerryzh168, kirklandsign Differential Revision: D49438891 fbshipit-source-id: a5cf5186d4a724cd3316c72f572da661b182ea20
diff --git a/examples/recipes/xnnpack_optimization/models.py b/examples/recipes/xnnpack_optimization/models.py
@@ -24,4 +24,5 @@ class OptimizationOptions(object):
     ),  # TODO[T163161310]: takes a long time to export to exec prog and save inception_v4 quantized model
     "w2l": OptimizationOptions(False, True),
     "dl3": OptimizationOptions(True, False),
+    "vit": OptimizationOptions(True, False),
 }

Original file line number	Diff line number	Diff line change
`@@ -24,4 +24,5 @@ class OptimizationOptions(object):`
`24`	`24`	`), # TODO[T163161310]: takes a long time to export to exec prog and save inception_v4 quantized model`
`25`	`25`	`"w2l": OptimizationOptions(False, True),`
`26`	`26`	`"dl3": OptimizationOptions(True, False),`
	`27`	`+ "vit": OptimizationOptions(True, False),`
`27`	`28`	`}`