From 23a104a4feec230d4d2337aebbc74ec6d8ea180a Mon Sep 17 00:00:00 2001 From: Henk Date: Sat, 15 Jul 2023 19:42:26 +0200 Subject: [PATCH] Only show 4-bit toggle on valid model --- modeling/inference_models/generic_hf_torch/class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modeling/inference_models/generic_hf_torch/class.py b/modeling/inference_models/generic_hf_torch/class.py index e871bbab..49b2b321 100644 --- a/modeling/inference_models/generic_hf_torch/class.py +++ b/modeling/inference_models/generic_hf_torch/class.py @@ -28,7 +28,7 @@ class model_backend(HFTorchInferenceModel): def get_requested_parameters(self, model_name, model_path, menu_path, parameters = {}): requested_parameters = super().get_requested_parameters(model_name, model_path, menu_path, parameters) dependency_exists = importlib.util.find_spec("bitsandbytes") - if dependency_exists: + if dependency_exists and (model_name != 'customhuggingface' or "custom_model_name" in parameters): requested_parameters.append({ "uitype": "toggle", "unit": "bool",