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",