Upload CondViTForEmbedding

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "CondViT-B16-cat",
+  "_name_or_path": "__debug_save",
   "architectures": [
     "CondViTForEmbedding"
   ],

hf_model.py

@@ -43,5 +43,5 @@ class CondViTForEmbedding(PreTrainedModel):
             n_categories=config.n_categories,
         )
 
-    def forward(self, img_tensors, category_indices=None):
-        return self.model(img_tensors, category_indices)
+    def forward(self, pixel_values, category_indices=None):
+        return self.model(imgs=pixel_values, c=category_indices)