Upload modeling_conceptframemet.py with huggingface_hub

modeling_conceptframemet.py

@@ -302,7 +302,19 @@ class ConceptFrameMetForMetaphorDetection(nn.Module):
         weights_path = os.path.join(model_path, "pytorch_model.bin")
         if os.path.exists(weights_path):
             state_dict = torch.load(weights_path, map_location='cpu')
-            model.load_state_dict(state_dict, strict=False)
+            
+            # Filter out source_qa_model and frame_qa_model weights if model doesn't have them
+            filtered_state_dict = {}
+            for key, value in state_dict.items():
+                # Skip source_qa_model weights if we don't have it
+                if key.startswith('source_qa_model.') and not model.has_source_predictor:
+                    continue
+                # Skip frame_qa_model weights if we don't have it  
+                if key.startswith('frame_qa_model.') and not model.has_frame_predictor:
+                    continue
+                filtered_state_dict[key] = value
+            
+            model.load_state_dict(filtered_state_dict, strict=False)
         
         return model