Update modeling_chatglm.py

modeling_chatglm.py

@@ -762,9 +762,8 @@ class GLMTransformer(torch.nn.Module):
     ):
         if not kv_caches:
             kv_caches = [None for _ in range(self.num_layers)]
-        else:
-            print("kv_caches",kv_caches)
-            kv_caches=kv_caches[1]
+
+        print("kv_caches",kv_caches)
         presents = () if use_cache else None
         if self.gradient_checkpointing and self.training:
             if use_cache: