Skip to content

Commit 6d3ac9a

Browse files
committed
[Code] Fix flake8 Lint errors
1 parent cd460ab commit 6d3ac9a

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

convert_hf_to_gguf.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4411,9 +4411,11 @@ def prepare_tensors(self):
44114411
if len(experts) > 0:
44124412
raise ValueError(f"Unprocessed experts: {experts}")
44134413

4414+
44144415
@Model.register("PLMForCausalLM")
44154416
class PLMModel(Model):
44164417
model_arch = gguf.MODEL_ARCH.PLM
4418+
44174419
def set_vocab(self):
44184420
self._set_vocab_gpt2()
44194421

@@ -4432,6 +4434,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
44324434
def prepare_tensors(self):
44334435
super().prepare_tensors()
44344436

4437+
44354438
@Model.register("T5WithLMHeadModel")
44364439
@Model.register("T5ForConditionalGeneration")
44374440
@Model.register("MT5ForConditionalGeneration")

gguf-py/gguf/constants.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1466,14 +1466,14 @@ class MODEL_TENSOR(IntEnum):
14661466
MODEL_TENSOR.FFN_EXP_PROBS_B,
14671467
],
14681468
MODEL_ARCH.PLM: [
1469-
MODEL_TENSOR.TOKEN_EMBD,
1469+
MODEL_TENSOR.TOKEN_EMBD,
14701470
MODEL_TENSOR.OUTPUT,
14711471
MODEL_TENSOR.OUTPUT_NORM,
14721472
MODEL_TENSOR.ATTN_NORM,
14731473
MODEL_TENSOR.ATTN_Q,
14741474
MODEL_TENSOR.ATTN_KV_A_MQA,
14751475
MODEL_TENSOR.ATTN_KV_A_NORM,
1476-
MODEL_TENSOR.ATTN_KV_B,
1476+
MODEL_TENSOR.ATTN_KV_B,
14771477
MODEL_TENSOR.ATTN_OUT,
14781478
MODEL_TENSOR.FFN_NORM,
14791479
MODEL_TENSOR.FFN_UP,

0 commit comments

Comments
 (0)