@@ -88,9 +88,7 @@ def export(self) -> "Lfm2p5VlEdgeManager":
8888 return self
8989
9090
91- def export_image_encoder (
92- lfm2 , quantize : bool = False
93- ) -> torch .export .ExportedProgram :
91+ def export_image_encoder (lfm2 , quantize : bool = False ) -> torch .export .ExportedProgram :
9492 """Export vision encoder as 'vision_encoder' method.
9593
9694 Input: [1, 3, 512, 512] float32 NCHW pixels in [0, 255]
@@ -120,9 +118,7 @@ def forward(self, images: torch.Tensor) -> torch.Tensor:
120118
121119 if quantize :
122120 logging .info ("Exporting vision encoder (int8 dynamic quantized)..." )
123- quantizer = XNNPACKQuantizer ().set_global (
124- get_symmetric_quantization_config ()
125- )
121+ quantizer = XNNPACKQuantizer ().set_global (get_symmetric_quantization_config ())
126122 manager = (
127123 Lfm2p5VlEdgeManager (
128124 model = encoder ,
@@ -285,7 +281,7 @@ def export_all(
285281 lfm2 = lfm2 .to (dtype .to_torch_dtype ())
286282
287283 logging .info ("[1/3] Exporting vision encoder..." )
288- vision_ep = export_image_encoder (lfm2 , quantize )
284+ vision_ep = export_image_encoder (lfm2 , quantize = False )
289285
290286 # Text decoder MUST come before token embedding (see export_token_embedding docstring)
291287 logging .info ("[2/3] Exporting text decoder..." )
0 commit comments