chore: Remove the jit implementation of layout predictor

Signed-off-by: Nikos Livathinos <[email protected]>
DS4SD · Dec 4, 2024 · c1bed7d · c1bed7d
1 parent fe8ef52
commit c1bed7d
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 185 deletions.
diff --git a/demo/demo_layout_predictor.py b/demo/demo_layout_predictor.py
@@ -14,11 +14,8 @@
 from huggingface_hub import snapshot_download
 from PIL import Image, ImageDraw, ImageFont
 
-# TODO: Switch LayoutModel implementations
 from docling_ibm_models.layoutmodel.layout_predictor import LayoutPredictor
 
-# from docling_ibm_models.layoutmodel.layout_predictor_jit import LayoutPredictor
-
 
 def save_predictions(prefix: str, viz_dir: str, img_fn: str, img, predictions: dict):
     img_path = Path(img_fn)
@@ -88,9 +85,6 @@ def demo(
 
             # Save predictions
             logger.info("Saving prediction visualization in: '%s'", viz_dir)
-
-            # TODO: Switch LayoutModel implementations
-            # save_predictions("JIT", viz_dir, img_fn, image, preds)
             save_predictions("ST", viz_dir, img_fn, image, preds)
     total_ms = 1000 * (time.perf_counter() - t0)
     avg_ms = (total_ms / img_counter) if img_counter > 0 else 0

diff --git a/docling_ibm_models/layoutmodel/layout_predictor_jit.py b/docling_ibm_models/layoutmodel/layout_predictor_jit.py