Skip to content

Commit b43c22e

Browse files
bug(medcat): CU-869cqnmtk Fix RelCAT typing issue (#388)
* CU-869cqnmtk: Fix for bert models * CU-869cqnmtk: Fix for llama models * CU-869cqnmtk: Fix for modern bert models --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
1 parent 2fabac9 commit b43c22e

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

medcat-v2/medcat/components/addons/relation_extraction/bert/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def load_specific(cls, pretrained_model_name_or_path: str,
8484
logger.info("Loaded model from pretrained: %s",
8585
pretrained_model_name_or_path)
8686
else:
87-
pretrained_model = cls.pretrained_model_name_or_path
87+
pretrained_model = model.pretrained_model_name_or_path
8888
model.hf_model = BertModel.from_pretrained(
8989
pretrained_model_name_or_path=pretrained_model,
9090
config=model_config.hf_model_config,

medcat-v2/medcat/components/addons/relation_extraction/llama/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ def load_specific(cls, pretrained_model_name_or_path: str,
218218
config=model_config.hf_model_config,
219219
ignore_mismatched_sizes=True, **kwargs)
220220
logger.info("Loaded model from pretrained: %s",
221-
cls.pretrained_model_name_or_path)
221+
model.pretrained_model_name_or_path)
222222

223223
return model
224224

medcat-v2/medcat/components/addons/relation_extraction/modernbert/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,6 @@ def load_specific(cls, pretrained_model_name_or_path: str,
8989
config=model_config.hf_model_config,
9090
ignore_mismatched_sizes=True, **kwargs)
9191
logger.info("Loaded model from pretrained: %s",
92-
cls.pretrained_model_name_or_path)
92+
model.pretrained_model_name_or_path)
9393

9494
return model

0 commit comments

Comments
 (0)