Skip to content

Commit 2c2abf2

Browse files
committed
fix formatting error
Signed-off-by: Mamta Singh <mamtsing@qti.qualcomm.com>
1 parent fbb85c0 commit 2c2abf2

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

QEfficient/base/modeling_qeff.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff 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():

0 commit comments

Comments
 (0)