Upload modeling_conceptframemet.py with huggingface_hub

modeling_conceptframemet.py

@@ -68,13 +68,14 @@ class AdaptiveSourceQAMelBert(nn.Module):
         self.source_alpha = getattr(args, 'source_alpha', 0.3)
         self.metaphor_threshold = getattr(args, 'metaphor_threshold', 0.5)
 
-        # Freeze or unfreeze source QA model
-        if not getattr(args, 'unfreeze_source_qa', False):
-            for param in self.source_qa_model.parameters():
-                param.requires_grad = False
-        else:
-            for param in self.source_qa_model.parameters():
-                param.requires_grad = True
+        # Freeze or unfreeze source QA model (only if it exists)
+        if self.source_qa_model is not None:
+            if not getattr(args, 'unfreeze_source_qa', False):
+                for param in self.source_qa_model.parameters():
+                    param.requires_grad = False
+            else:
+                for param in self.source_qa_model.parameters():
+                    param.requires_grad = True
 
         # Load source labels
         self.source_id2label = {}