haflway through modularizing

pyg-team · Jul 10, 2024 · e131d53 · e131d53
1 parent 5badc10
commit e131d53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_geometric/nn/nlp/llm.py b/torch_geometric/nn/nlp/llm.py
@@ -172,6 +172,7 @@ def pad_embeds(batch_inputs_embeds, batch_attention_mask, batch_label_input_ids=
             label_input_ids = None
         return inputs_embeds, attention_mask, label_input_ids
 
+    def _get_embeds
     def forward(
         self,
         question: List[str],