From 6d28c381d2713861fa04a2a4deb121c3ed1a560e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= <me@danieldk.eu>
Date: Tue, 16 Jul 2024 06:33:00 +0000
Subject: [PATCH] Remove stray `quantize` argument in
 `get_weights_col_packed_qkv`

Fixes #2236.
---
 server/text_generation_server/utils/weights.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/server/text_generation_server/utils/weights.py b/server/text_generation_server/utils/weights.py
index 1c55dd74..b530af23 100644
--- a/server/text_generation_server/utils/weights.py
+++ b/server/text_generation_server/utils/weights.py
@@ -261,7 +261,6 @@ class Weights:
     def get_weights_col_packed_qkv(
         self,
         prefix: str,
-        quantize: str,
         num_heads: int,
         num_key_value_heads: int,
     ):