Upload model

config.json

@@ -36,6 +36,6 @@
   "resolution": 560,
   "sa_nheads": 8,
   "torch_dtype": "float32",
-  "transformers_version": "4.51.1",
+  "transformers_version": "4.50.3",
   "two_stage": true
 }

modeling_rf_detr.py

@@ -39,14 +39,14 @@ class RFDetrModelForObjectDetection(PreTrainedModel):
             lite_refpoint_refine = config.lite_refpoint_refine,
             layer_norm = config.layer_norm,
             amp = config.amp,
-            #num_classes = config.num_classes,
+            num_classes = config.num_classes,
             device = config.device,
             resolution = config.resolution,
             group_detr = config.group_detr,
             gradient_checkpointing = config.gradient_checkpointing,
             num_queries = config.num_queries,
             encoder = config.encoder,
-            #hidden_dim = config.hidden_dim,
+            hidden_dim = config.hidden_dim,
             sa_nheads = config.sa_nheads,
             ca_nheads = config.ca_nheads,
             dec_n_points = config.dec_n_points,