remove num_labels arg

unitaryai · Oct 23, 2023 · ec3ea57 · ec3ea57
1 parent 754c119
commit ec3ea57
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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,
     )