Update backends/smoldocling.py

backends/smoldocling.py

@@ -12,7 +12,7 @@ from PIL import Image
 from transformers import AutoModelForVision2Seq, AutoProcessor
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-MAX_PAGES = 2
+MAX_PAGES = 1
 
 # Initialize processor and model
 processor = AutoProcessor.from_pretrained("ds4sd/SmolDocling-256M-preview")
@@ -49,7 +49,7 @@ def convert_smoldocling(path: str, file_name: str):
         inputs = inputs.to(DEVICE)
 
         # Generate outputs
-        generated_ids = model.generate(**inputs, max_new_tokens=8192 * 2)
+        generated_ids = model.generate(**inputs, max_new_tokens=8192)
         prompt_length = inputs.input_ids.shape[1]
         trimmed_generated_ids = generated_ids[:, prompt_length:]
         doctags = processor.batch_decode(