Have you tried to post this on the Hugginface forum?
As it seems that you are using one of their classes TFAutoModelForSequenceClassification
Edit:
I suppose your post is this one:
  
      

      Hugging Face Forums – 21 Jan 22
  

  
    

Save a tensorflow model with a transformer layer 3

  Hi  I trained a model with the following architecture:  bert_config = BertConfig.from_pretrained(MODEL_NAME) bert_config.output_hidden_states = True backbone = TFAutoModelForSequenceClassification.from_pretrained(MODEL_NAME,config=bert_config) ...