diff --git a/neural_speed/convert/convert_mistral.py b/neural_speed/convert/convert_mistral.py index 89d359f62..949675df5 100644 --- a/neural_speed/convert/convert_mistral.py +++ b/neural_speed/convert/convert_mistral.py @@ -1062,7 +1062,7 @@ def write_file_header(self, params: Params, file_type: NEFileType) -> None: self.fout.write(struct.pack("i", 0)) self.fout.write(struct.pack("i", params.ffn_hidden_size)) self.fout.write(struct.pack("i", 0)) - + self.fout.write(struct.pack("i", 0)) self.fout.write(struct.pack("i", 0)) self.fout.write(struct.pack("f", params.rms_norm_eps))