Upload folder using huggingface_hub

modeling_helmbert.py

@@ -559,7 +559,7 @@ class HELMBertEncoder(nn.Module):
         all_attentions = () if output_attentions else None
 
         # Store for nGiE
-        ngie_input_states = hidden_states.clone()
+        ngie_input_states = hidden_states
 
         # [batch, seq, hidden] -> [seq, batch, hidden]
         hidden_states = hidden_states.transpose(0, 1)
@@ -594,7 +594,7 @@ class HELMBertEncoder(nn.Module):
 
             # Store layer[-2] for EMD
             if use_emd and layer_idx == num_layers - 2:
-                layer_minus_2 = hidden_states.clone()
+                layer_minus_2 = hidden_states
 
         # Convert back to [batch, seq, hidden]
         hidden_states = hidden_states.transpose(0, 1)