Skip to content

Commit f78c724

Browse files
committed
fix python lint
1 parent cc468d7 commit f78c724

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

convert_hf_to_gguf.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5288,8 +5288,7 @@ def __init__(self, *args, **kwargs):
52885288
mod_conf = json.load(mod_conf_json_file)
52895289
# hparams dense_2_feat_out and dense_3_feat_in are required when loading model's dense weights
52905290
prefix = self._get_dense_prefix(mod_path)
5291-
if (mod_conf["in_features"] is not None
5292-
and mod_conf["out_features"] is not None):
5291+
if mod_conf["in_features"] is not None and mod_conf["out_features"] is not None:
52935292
self.dense_features_dims[prefix] = (mod_conf["in_features"], mod_conf["out_features"])
52945293

52955294
def generate_extra_tensors(self) -> Iterable[tuple[str, Tensor]]:
@@ -5329,8 +5328,8 @@ def set_gguf_parameters(self):
53295328
self.gguf_writer.add_sliding_window(orig_sliding_window)
53305329
if self.sentence_transformers_dense_modules:
53315330
for dense, dims in self.dense_features_dims.items():
5332-
logger.info(f"Setting dense layer {dense} in/out features to {dims}")
5333-
self.gguf_writer.add_dense_features_dims(dense, dims[0], dims[1])
5331+
logger.info(f"Setting dense layer {dense} in/out features to {dims}")
5332+
self.gguf_writer.add_dense_features_dims(dense, dims[0], dims[1])
53345333

53355334
self._try_set_pooling_type()
53365335

0 commit comments

Comments
 (0)