alfiannajih
/

g-retriever-resume-reviewer

Text Generation

knowledge graph

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

alfiannajih commited on Aug 21, 2024

Commit

06fa6b1

·

verified ·

1 Parent(s): db7044b

Update g_retriever.py

Files changed (1) hide show

g_retriever.py +1 -1

g_retriever.py CHANGED Viewed

@@ -42,7 +42,7 @@ class GRetrieverModel(LlamaForCausalLM):
         # mean pooling
         g_embeds = global_mean_pool(n_embeds, graph.batch.to(n_embeds.device))
-        return g_embeds.to(model.device)
     @wraps(LlamaForCausalLM.forward)
     def forward(

         # mean pooling
         g_embeds = global_mean_pool(n_embeds, graph.batch.to(n_embeds.device))
+        return g_embeds.to(self.model.device)
     @wraps(LlamaForCausalLM.forward)
     def forward(