Upload model

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,