Model not getting saved after fine tuning.
#7
by
samant
- opened
When i am fine tuning this model on my dataset, only three files are getting but not the actual model.
Files:
config.json
generation_config.json
model.safetensors
Please help me regarding the same.
Those are the model files that are needed. The only files that are missing are the tokenizer files. Did you make sure to save the tokenizer?
Yes i have used " model.save_pretrained(args.save_dir) ". This should should save tokenizers as well right?
Can you please guide me how to use this model for prediction purpose?