How to use trainer.train() with OVModelForCausalLM() model

[ctrivino1]

I am currently converting a local LLM to Open Vino, I would like to fine tune my model with the Trainer function but I get an error stating: AttributeError: 'OVModelForCausalLM' object has no attribute 'named_children'

Please let me know if there is a way to fine tune openVino models that are loaded with OVModelForCausalLM().

Attached is my script
Fine_Tuning_mistral_7b_v3 (2).zip