diff --git a/magic_pdf/model/pdf_extract_kit.py b/magic_pdf/model/pdf_extract_kit.py index 406417cd..7f9c6808 100644 --- a/magic_pdf/model/pdf_extract_kit.py +++ b/magic_pdf/model/pdf_extract_kit.py @@ -64,6 +64,7 @@ def mfr_model_init(weight_dir, cfg_path, _device_='cpu'): task = tasks.setup_task(cfg) model = task.build_model(cfg) model = model.to(_device_) + model.eval() vis_processor = load_processor('formula_image_eval', cfg.config.datasets.formula_rec_eval.vis_processor.eval) mfr_transform = transforms.Compose([vis_processor, ]) return [model, mfr_transform]