From 84a942ef666f2996924008be2273c874c3702ceb Mon Sep 17 00:00:00 2001 From: Zhao-Penghai Date: Thu, 28 Nov 2024 12:03:12 +0800 Subject: [PATCH] Update NAIP_LLaMA.py --- llama_for_naip/NAIP_LLaMA.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/llama_for_naip/NAIP_LLaMA.py b/llama_for_naip/NAIP_LLaMA.py index 4112818..d6e1dd0 100644 --- 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]