Skip to content

Commit 7473f01

Browse files
committed
Fix is_external_encoder to check for cnlpt models even on hub (simplifies logic).
1 parent 62d4663 commit 7473f01

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/cnlpt/train_system.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def is_external_encoder(model_name_or_path: str) -> bool:
152152
:param model_name_or_path: specified model
153153
:return: whether the encoder is an external (non-cnlpt) model
154154
"""
155-
return is_hub_model(model_name_or_path) or not is_cnlpt_model(model_name_or_path)
155+
return not is_cnlpt_model(model_name_or_path)
156156

157157

158158
def main(

0 commit comments

Comments
 (0)