Skip to content

Commit 4b4975c

Browse files
author
pockers21
committed
fix: resolve code formatting issues
1 parent 87daf40 commit 4b4975c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

convert_hf_to_gguf.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2711,6 +2711,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
27112711
return []
27122712
yield from super().modify_tensors(data_torch, name, bid)
27132713

2714+
27142715
@ModelBase.register("Eagle2DraftForCausalLM")
27152716
class Eagle2DraftModel(TextModel):
27162717
model_arch = gguf.MODEL_ARCH.EAGLE2_DRAFT
@@ -2729,6 +2730,7 @@ def set_gguf_parameters(self):
27292730
self.gguf_writer.add_rope_scaling_factor(self.hparams["rope_scaling"]["factor"])
27302731
self.gguf_writer.add_rope_scaling_orig_ctx_len(self.hparams["rope_scaling"]["original_max_position_embeddings"])
27312732

2733+
27322734
@ModelBase.register(
27332735
"Qwen2VLModel",
27342736
"Qwen2VLForConditionalGeneration",

gguf-py/gguf/tensor_mapping.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ class TensorNameMap:
5858
"wpe", # gpt2
5959
),
6060

61-
#eagle2 draft model
61+
# eagle2 draft model
6262
MODEL_TENSOR.FC: (
63-
"model.fc",
63+
"model.fc",
6464
),
6565
# Output
6666
MODEL_TENSOR.OUTPUT: (

0 commit comments

Comments
 (0)