Workaround HPU dependency on specific transformers version

Signed-off-by: Sergey Plotnikov <[email protected]>

Author: splotnikv

src/training_hub/profiling/memory_estimator.py

@@ -1,6 +1,5 @@
 from typing import override
 from transformers import AutoModel
-from mini_trainer.osft_utils import MODEL_CONFIGS
 import numpy as np
 
 """
@@ -322,6 +321,9 @@ def __init__(
             raise ValueError("Ratio must be in the range [0, 1]")
 
         # Check to see which terms need to be included in the search for valid layers
+        # MODEL_CONFIGS require transformers>4.56.0, that conflict with HPU enabling,
+        # temporary moving MODEL_CONFIGS import here to enable HPU support
+        from mini_trainer.osft_utils import MODEL_CONFIGS
         self.target_terms = MODEL_CONFIGS['default']['patterns']
         for key in MODEL_CONFIGS.keys():
             if self.model_path.find(key) > -1: