Latest Changes

core/model_loader.py

@@ -37,19 +37,15 @@ def generate_caption(
 
     # Style-specific decoding
     if style == "Image Explanation":
-        max_len = 140
         min_len = 80
     elif style == "Detailed Caption":
-        max_len = 90
         min_len = 50
     else:
-        max_len = 40
         min_len = 20
 
     with torch.inference_mode():
         output_ids = model.generate(
             **inputs,
-            max_length=max_len,
             min_length=min_len,
             do_sample=True,
             top_p=0.9,