NagaSaiAbhinay
/

CSD

Feature Extraction

Model card Files Files and versions

NagaSaiAbhinay commited on Jan 12

Commit

2ffab58

·

verified ·

1 Parent(s): 0dea13a

Update model.py

Files changed (1) hide show

model.py +2 -1

model.py CHANGED Viewed

@@ -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)

     def forward(self, pixel_values):
         features = self.backbone(pixel_values)
         style_embeds = self.out_style(features)
+        content_embeds = self.out_content(features)
+        return features, style_embeds, content_embeds