py : whitespaces

This commit is contained in:
Georgi Gerganov 2023-12-16 11:01:02 +02:00
parent aa5c881adb
commit 7500fa2f07
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -984,6 +984,7 @@ class QwenModel(Model):
print(f"{new_name}, n_dims = {n_dims}, {old_dtype} --> {data.dtype}")
self.gguf_writer.add_tensor(new_name, data)
class Phi2Model(Model):
def set_gguf_parameters(self):
block_count = self.hparams["n_layer"]
@ -999,6 +1000,7 @@ class Phi2Model(Model):
self.gguf_writer.add_rope_dimension_count(self.hparams["rotary_dim"])
self.gguf_writer.add_file_type(self.ftype)
###### CONVERSION LOGIC ######