Bug fix: Assigning to_device only in non model agnostic case and also assignment of text_similarity tokenizzer and model in model agnostic case
Bug fix: Assigning to_device only in non model agnostic case and also assignment of text_similarity tokenizzer and model in model agnostic case

slundberg/shap,slundberg/shap,slundberg/shap,slundberg/shap