Upload modeling_minicpm_reranker.py

modeling_minicpm_reranker.py

@@ -1265,6 +1265,8 @@ class LayerWiseMiniCPMForCausalLM(MiniCPMPreTrainedModel):
             )
 
         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}]")
 
         # decoder outputs consists of (dec_features, layer_state, dec_hidden, dec_attn)
         outputs = self.model(