Fix for custom model names

2025-06-05 21:59:24 +02:00 · 2023-05-26 21:40:39 -04:00
parent 9bc9021843
commit cce5c1932c
3 changed files with 6 additions and 2 deletions
--- a/modeling/inference_models/hf.py
+++ b/modeling/inference_models/hf.py
@@ -188,6 +188,7 @@ class HFInferenceModel(InferenceModel):
            self.usegpu = parameters['use_gpu'] if 'use_gpu' in parameters else None
            self.breakmodel = False
            self.lazy_load = False
+        logger.info(parameters)
        self.model_name = parameters['custom_model_name'] if 'custom_model_name' in parameters else parameters['id']
        self.path = parameters['path'] if 'path' in parameters else None