Skip to content

Commit bf72ac3

Browse files
committed
build: Removed huggingface_hub import + error check from run_quant.py
Signed-off-by: Brandon Groth <[email protected]>
1 parent f10b56a commit bf72ac3

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

fms_mo/run_quant.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@
3434

3535
# Third Party
3636
from datasets import load_from_disk
37-
from huggingface_hub.errors import HFValidationError
3837
from torch.cuda import OutOfMemoryError
3938
from transformers import AutoTokenizer
4039
import 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(

0 commit comments

Comments
 (0)