bugfix

Author: voorhs

autointent/_optimization_config.py

@@ -25,7 +25,7 @@ class OptimizationConfig(BaseModel):
 
     cross_encoder_config: CrossEncoderConfig = CrossEncoderConfig()
 
-    hf_model_config: HFModelConfig = HFModelConfig()
+    transformer_config: HFModelConfig = HFModelConfig()
 
     sampler: SamplerType = "brute"
     """See tutorial on optuna and presets."""

autointent/_pipeline/_pipeline.py

@@ -68,7 +68,7 @@ def __init__(
             self.embedder_config = EmbedderConfig()
             self.cross_encoder_config = CrossEncoderConfig()
             self.data_config = DataConfig()
-            self.hf_model_config = HFModelConfig()
+            self.transformer_config = HFModelConfig()
         elif not isinstance(nodes[0], InferenceNode):
             assert_never(nodes)
 
@@ -139,7 +139,7 @@ def from_optimization_config(cls, config: dict[str, Any] | Path | str | Optimiza
         pipeline.set_config(optimization_config.data_config)
         pipeline.set_config(optimization_config.embedder_config)
         pipeline.set_config(optimization_config.cross_encoder_config)
-        pipeline.set_config(optimization_config.hf_model_config)
+        pipeline.set_config(optimization_config.transformer_config)
         return pipeline
 
     def _fit(self, context: Context, sampler: SamplerType) -> None: