enable vit quantization

kimishpatel · facebook-github-bot · commit 4d700c46d168 · 2023-09-19T21:48:26.000-07:00
Summary: After 0919 nightly of pytorch vit linear layers should be quantized

Differential Revision: D49438891
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`	`}`