neuml / txtinstruct

📚 Datasets and models for instruction-tuning
Apache License 2.0
233 stars 11 forks source link

Save model? #4

Open ranjancse26 opened 1 year ago

ranjancse26 commented 1 year ago

Review txtinstruct is a great library for instruction fine tuning.

Just wondering about saving the instruction fine-tuned model? Please help me with this.

davidmezzetti commented 1 year ago

Thank you for the kind words.

I'm assuming you're talking about how to save the fine-tuned model created by the Instructor class. In that case, you can pass an output_dir directory to the call. More details on the available options can be found here.

ranjancse26 commented 1 year ago

Thanks @davidmezzetti I am now able to save the model.

image