fix: add kwargs to model to ignore additional arguments

model.py

@@ -31,6 +31,6 @@ class CSDModel(PreTrainedModel):
         )
 
     @torch.inference_mode()
-    def forward(self, pixel_values: torch.Tensor) -> CSDOutput:
+    def forward(self, pixel_values: torch.Tensor, **kwargs) -> CSDOutput:
         image_embeds, style_embeds, content_embeds = self.model(pixel_values)
         return CSDOutput(image_embeds=image_embeds, style_embeds=style_embeds, content_embeds=content_embeds)