update model_class.py to be compatible with new transformers (transformers==4.36.2) e8e231a verified prompteus commited on Mar 13