Update NAIP_LLaMA.py

ssocean · Nov 28, 2024 · 84a942e · 84a942e
1 parent 411d60a
commit 84a942e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/llama_for_naip/NAIP_LLaMA.py b/llama_for_naip/NAIP_LLaMA.py
@@ -61,6 +61,9 @@ def forward(
         )
         hidden_states = transformer_outputs[0]
         logits = self.score(hidden_states)
+        # modify here
+        logits = self.sigmoid(logits)
+
 
         if input_ids is not None:
             batch_size = input_ids.shape[0]