refactor: move image processors to separate files (#4229)

2025-03-11 12:35:35 -07:00
parent 0f2a2e3c19
commit ff2ce0b86f
22 changed files with 1085 additions and 955 deletions
@@ -202,7 +202,7 @@ class MllamaVisionEncoderLayer(nn.Module):
            quant_config=None,
            dropout=0.0,
            use_context_forward=False,
-            use_full_precision_softmax=False,
+            softmax_in_single_precision=False,
            flatten_batch=False,
            prefix=add_prefix("self_attn", prefix),
        )