From 2c681e4ac219bd49e0e59db594ebd0f390a8524e Mon Sep 17 00:00:00 2001 From: JegernOUTT Date: Mon, 15 Jan 2024 15:47:01 +1030 Subject: [PATCH] missing exception block --- self_hosting_machinery/finetune/scripts/finetune_filter.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/self_hosting_machinery/finetune/scripts/finetune_filter.py b/self_hosting_machinery/finetune/scripts/finetune_filter.py index 3eb0b66c..8becedc9 100644 --- a/self_hosting_machinery/finetune/scripts/finetune_filter.py +++ b/self_hosting_machinery/finetune/scripts/finetune_filter.py @@ -108,6 +108,9 @@ def _get_file_loss(file) -> float: except InvalidLossValueException as e: files_status_context.reject_file(file, reason=str(e)) continue + except Exception as e: + files_status_context.reject_file(file, reason=str(e)) + continue if file_loss > filter_loss_threshold: files_status_context.reject_file(file, reason=f"loss {file_loss:.3f}")