Skip to content

Commit f88325e

Browse files
committed
feat: upgrade response template warning to error
Signed-off-by: ChanderG <[email protected]>
1 parent ea2eb7a commit f88325e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tuning/sft_trainer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import sys
2222
import time
2323
import traceback
24+
import warnings
2425

2526
# Third Party
2627
from huggingface_hub.utils._validators import HFValidationError
@@ -856,4 +857,5 @@ def main():
856857

857858

858859
if __name__ == "__main__":
860+
warnings.filterwarnings("error", "Could not find response key", UserWarning)
859861
main()

0 commit comments

Comments
 (0)