File tree Expand file tree Collapse file tree 1 file changed +9
-4
lines changed
Expand file tree Collapse file tree 1 file changed +9
-4
lines changed Original file line number Diff line number Diff line change @@ -295,9 +295,12 @@ def _export(
295295 self .onnx_path = onnx_path
296296 return onnx_path
297297
298- def get_onnx_path (self , prefill_only : Optional [bool ] = False ,
299- specializations : Optional [List [Dict [str , int ]]] = None ,
300- offload_pt_weights : Optional [bool ] = True ):
298+ def get_onnx_path (
299+ self ,
300+ prefill_only : Optional [bool ] = False ,
301+ specializations : Optional [List [Dict [str , int ]]] = None ,
302+ offload_pt_weights : Optional [bool ] = True ,
303+ ):
301304 kwargs = {"offload_pt_weights" : offload_pt_weights }
302305 if prefill_only :
303306 if self .prefill_onnx_path is None :
@@ -349,7 +352,9 @@ def _compile(
349352
350353 For QNN Compilation path, when enable_qnn is set to True, any parameter passed in compiler_options will be ignored.
351354 """
352- onnx_path = Path (onnx_path if onnx_path else self .get_onnx_path (prefill_only , specializations , offload_pt_weights ))
355+ onnx_path = Path (
356+ onnx_path if onnx_path else self .get_onnx_path (prefill_only , specializations , offload_pt_weights )
357+ )
353358 compile_dir = Path (compile_dir or onnx_path .parent )
354359 qpc_path = compile_dir / "qpc"
355360 if not onnx_path .is_file ():
You can’t perform that action at this time.
0 commit comments