Update g_retriever.py

g_retriever.py

@@ -63,7 +63,7 @@ class GRetrieverModel(LlamaForCausalLM):
         )
         return_dict = return_dict if return_dict is not None else self.config.use_return_dict
 
-        if inputs.shape != torch.Size([1, 1]):
+        if (inputs==-1).any():
             # embed bos prompt
             bos_embeds = self.get_input_embeddings()(torch.tensor(
                 self.config.bos_id,