Update model.py

model.py

@@ -15,4 +15,5 @@ class CSDModel(PreTrainedModel):
     def forward(self, pixel_values):
         features = self.backbone(pixel_values)
         style_embeds = self.out_style(features)
-        content_embeds = self.out_content(features)
+        content_embeds = self.out_content(features)
+        return features, style_embeds, content_embeds