Upload modeling_minicpm_reranker.py

modeling_minicpm_reranker.py

@@ -1261,12 +1261,12 @@ class LayerWiseMiniCPMForCausalLM(MiniCPMPreTrainedModel):
         remove_layers = [i for i in cutoff_layers if self.config.start_layer > i or i > self.config.num_hidden_layers]
         if len(remove_layers) > 0:
             logger.warning_once(
-                f"layers {remove_layers} is incompatible with the setting. They will be removed..."
+                f"layers {remove_layers} are incompatible with the setting. They will be removed..."
             )
 
         cutoff_layers = [i for i in cutoff_layers if i not in remove_layers]
         if len(cutoff_layers) == 0:
-            raise ValueError(f"Your cutoff layers must in [{self.config.start_layer}, {self.config.num_hidden_layers + 1}]")
+            raise ValueError(f"Your cutoff layers must in [{self.config.start_layer}, {self.config.num_hidden_layers}]")
 
         # decoder outputs consists of (dec_features, layer_state, dec_hidden, dec_attn)
         outputs = self.model(