Skip to content

Commit 5f0bec3

Browse files
quic-akuruvilAnn Kuruvilla
authored andcommitted
Re-Format
Signed-off-by: Ann Kuruvilla <akuruvil@qti.qualcomm.com>
1 parent 8d7d475 commit 5f0bec3

File tree

4 files changed

+19
-9
lines changed

4 files changed

+19
-9
lines changed

QEfficient/base/modeling_qeff.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -400,7 +400,7 @@ def _compile(
400400

401401
if mdp_ts_json_path := compiler_options.pop("mdp_load_partition_config", None):
402402
command.append(f"-mdp-load-partition-config={mdp_ts_json_path}")
403-
403+
404404
for key, value in compiler_options.items():
405405
option = "-" + key.replace("_", "-")
406406
if isinstance(value, bool):

QEfficient/transformers/models/gemma3/modeling_gemma3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -687,7 +687,7 @@ def get_npi_file(self, model_name: str, **compiler_options):
687687
f"For Model {self.pretrained_model_name_or_path} default NPI file is not supported/added. Please use one of the following: google/gemma-3-4b-it, google/gemma-3-27b-it"
688688
)
689689
return compiler_options
690-
690+
691691
def get_specializations(
692692
self,
693693
batch_size: int,

QEfficient/transformers/models/modeling_auto.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1178,7 +1178,7 @@ def compile(
11781178
self.export(
11791179
use_onnx_subfunctions=use_onnx_subfunctions,
11801180
)
1181-
1181+
11821182
# TODO this hould be removed once the continous batching is supported for all the models.
11831183
compiler_options.pop("continuous_batching", None)
11841184
compiler_options.pop("kv_cache_batch_size", None)
@@ -1197,11 +1197,11 @@ def compile(
11971197
use_onnx_subfunctions=use_onnx_subfunctions,
11981198
**compiler_options,
11991199
)
1200-
1201-
# Custom NPI file options
1200+
1201+
# Custom NPI file options
12021202
if hasattr(self.model, "get_npi_file"):
12031203
compiler_options = self.model.get_npi_file(self.model.name_or_path)
1204-
1204+
12051205
if not skip_lang:
12061206
custom_io_lang = {}
12071207
# Inputs
@@ -1802,7 +1802,7 @@ def compile(
18021802
img_size=img_size,
18031803
**compiler_options,
18041804
)
1805-
1805+
18061806
if hasattr(self.model, "get_npi_file"):
18071807
compiler_options = self.model.get_npi_file(self.model.name_or_path)
18081808

QEfficient/utils/constants.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,20 @@
2929

3030
# Gemma3 Constant
3131
DEFAULT_GEMMA3_4B_NODE_PRECISION_INFO = str(
32-
Path(__file__).resolve().parent.parent / "transformers" / "models" / "gemma3" / "configs" / "fp32_nodes_gemma3_4b.yaml"
32+
Path(__file__).resolve().parent.parent
33+
/ "transformers"
34+
/ "models"
35+
/ "gemma3"
36+
/ "configs"
37+
/ "fp32_nodes_gemma3_4b.yaml"
3338
)
3439
DEFAULT_GEMMA3_27B_NODE_PRECISION_INFO = str(
35-
Path(__file__).resolve().parent.parent / "transformers" / "models" / "gemma3" / "configs" / "fp32_nodes_gemma3_27b.yaml"
40+
Path(__file__).resolve().parent.parent
41+
/ "transformers"
42+
/ "models"
43+
/ "gemma3"
44+
/ "configs"
45+
/ "fp32_nodes_gemma3_27b.yaml"
3646
)
3747
# Compiler defaults
3848
DEFAULT_AIC_NUM_CORES = 16

0 commit comments

Comments
 (0)