py : fix lint (#4889)

This commit is contained in:
Georgi Gerganov 2024-01-12 13:03:38 +02:00
parent f445c0e68c
commit 2d00741e12
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -825,7 +825,7 @@ class PersimmonModel(Model):
# NOTE: not sure about this change - why does the model not have a rope dimension count when it is smaller # NOTE: not sure about this change - why does the model not have a rope dimension count when it is smaller
# than the head size? # than the head size?
# ref: https://github.com/ggerganov/llama.cpp/pull/4889 # ref: https://github.com/ggerganov/llama.cpp/pull/4889
#self.gguf_writer.add_rope_dimension_count(hidden_size // head_count) # self.gguf_writer.add_rope_dimension_count(hidden_size // head_count)
self.gguf_writer.add_rope_dimension_count(hidden_size // head_count // 2) self.gguf_writer.add_rope_dimension_count(hidden_size // head_count // 2)
self.gguf_writer.add_head_count(head_count) self.gguf_writer.add_head_count(head_count)