File tree Expand file tree Collapse file tree 1 file changed +0
-7
lines changed
Expand file tree Collapse file tree 1 file changed +0
-7
lines changed Original file line number Diff line number Diff line change 3434
3535# Third Party
3636from datasets import load_from_disk
37- from huggingface_hub .errors import HFValidationError
3837from torch .cuda import OutOfMemoryError
3938from transformers import AutoTokenizer
4039import torch
@@ -353,12 +352,6 @@ def main():
353352 logger .error (traceback .format_exc ())
354353 write_termination_log (f"Unable to load file: { e } " )
355354 sys .exit (USER_ERROR_EXIT_CODE )
356- except HFValidationError as e :
357- logger .error (traceback .format_exc ())
358- write_termination_log (
359- f"There may be a problem with loading the model. Exception: { e } "
360- )
361- sys .exit (USER_ERROR_EXIT_CODE )
362355 except (TypeError , ValueError , EnvironmentError ) as e :
363356 logger .error (traceback .format_exc ())
364357 write_termination_log (
You can’t perform that action at this time.
0 commit comments