-weightedmean-nli-bitfit") model = AutoModel.from_pretrained("Muennighoff/SGPT-125-weightedmean-nli-bitfit") # Deactivate Dropout (There is no dropout in the above models so it makes no difference here but other SGPT models may have dropout) model.eval() # Tokenize input texts texts=...