diff --git a/autopeptideml/reps/lms.py b/autopeptideml/reps/lms.py index f40d911..ee9b2a5 100644 --- a/autopeptideml/reps/lms.py +++ b/autopeptideml/reps/lms.py @@ -316,7 +316,6 @@ def _rep_batch( elif 'peptidemtr' in self.model_name.lower(): embd_rpr = self.model(**inputs)['last_layer'] else: - print(self.model(**inputs)) embd_rpr = self.model(**inputs).last_hidden_state output = [] for idx in range(len(batch)):