Ramesh_Kumar:

BertForSequenceClassification.from_pretrained


ah, I see.  In that case, one place to check would be BertForSequenceClassification.from_pretrained - it might be assuming a floating point model.  You would have to modify the loading code.