Update modeling_chatglm.py

modeling_chatglm.py

@@ -860,11 +860,10 @@ class ChatGLMModel(ChatGLMPreTrainedModel):
 
         all_self_attentions = None
         all_hidden_states = () if output_hidden_states else None
-        for index in range(self.num_layers):
+        for index, layer in enumerate(self.layers):
             if output_hidden_states:
                 all_hidden_states = all_hidden_states + (hidden_states,)
 
-            layer = self._get_layer(index)
             if self.gradient_checkpointing and self.training:
                 layer_ret = torch.utils.checkpoint.checkpoint(
                     layer,