FreedomIntelligence / AceGPT

Apache License 2.0
106 stars 6 forks source link

model.save_pretrained not working for AceGPT #15

Open ashmalvayani opened 4 months ago

ashmalvayani commented 4 months ago

Hello, I am trying to simply load the model and save it again using _model.savepretrained method but it's throwing the following error (As shown in the image below). I have searched a bit at it looks like there's some error with generation_config.json file.

My current transformer version is 4.39.0.dev0. When I change it to 4.36.0 it starts saving it as safe_tensors. Do you think is that a version with the transformers version or the generation_config file needs to be deleted to run the code?

image