From f652788d543098c687ee4b1eb7903b7f49d48ec0 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Thu, 1 Jun 2023 11:22:19 +0200
Subject: [PATCH] fix

---
 server/text_generation_server/models/flash_santacoder.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/server/text_generation_server/models/flash_santacoder.py b/server/text_generation_server/models/flash_santacoder.py
index 6ff8286a..2a747f50 100644
--- a/server/text_generation_server/models/flash_santacoder.py
+++ b/server/text_generation_server/models/flash_santacoder.py
@@ -90,7 +90,7 @@ class FlashSantacoder(FlashCausalLM):
                     filename, framework="pt", device=str(device) if quantize is None else "cpu"
             ) as f:
                 for key in f.keys():
-                    value = f.get_slice(key)
+                    value = f.get_tensor(key)
                     value = value.to(device if quantize is None else "cpu").to(dtype)
 
                     layer_name = ".".join(key.split(".")[:4])