fixed warning

Signed-off-by: Matteo Omenetti <[email protected]>
DS4SD · Feb 3, 2025 · e53c4c5 · e53c4c5
1 parent d13cb50
commit e53c4c5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/docling_ibm_models/code_formula_model/models/sam_opt_image_processor.py b/docling_ibm_models/code_formula_model/models/sam_opt_image_processor.py
@@ -28,4 +28,7 @@ def __call__(self, image):
         return image
 
 
-AutoImageProcessor.register(SamOptImageProcessor, SamOptImageProcessor)
+AutoImageProcessor.register(
+    config_class="SamOptImageProcessor",
+    slow_image_processor_class=SamOptImageProcessor,
+)