From ec3ea57a66f5553286e47e760e15cd25f554d52e Mon Sep 17 00:00:00 2001
From: Laura Hanu <32672979+laurahanu@users.noreply.github.com>
Date: Tue, 24 Oct 2023 00:39:38 +0100
Subject: [PATCH] remove num_labels arg

---
 detoxify/detoxify.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/detoxify/detoxify.py b/detoxify/detoxify.py
index 8147df3..6d0c71a 100644
--- a/detoxify/detoxify.py
+++ b/detoxify/detoxify.py
@@ -23,7 +23,6 @@ def get_model_and_tokenizer(
     model = model_class.from_pretrained(
         pretrained_model_name_or_path=None,
         config=huggingface_config_path or config,
-        num_labels=num_classes,
         state_dict=state_dict,
         local_files_only=huggingface_config_path is not None,
     )