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

Commit

0308adf

•

1 Parent(s): ca0a58b

Update g_retriever.py

Files changed (1) hide show

g_retriever.py +0 -1

g_retriever.py CHANGED Viewed

@@ -67,7 +67,6 @@ class GRetrieverModel(LlamaForCausalLM):
             # embed bos prompt
             bos_embeds = self.get_input_embeddings()(torch.tensor(
                 self.config.bos_id,
-                dtype=self.model.dtype,
                 device=self.model.device
             ))

             # embed bos prompt
             bos_embeds = self.get_input_embeddings()(torch.tensor(
                 self.config.bos_id,
                 device=self.model.device
             ))