alfiannajih commited on
Commit
0308adf
1 Parent(s): ca0a58b

Update g_retriever.py

Browse files
Files changed (1) hide show
  1. g_retriever.py +0 -1
g_retriever.py CHANGED
@@ -67,7 +67,6 @@ class GRetrieverModel(LlamaForCausalLM):
67
  # embed bos prompt
68
  bos_embeds = self.get_input_embeddings()(torch.tensor(
69
  self.config.bos_id,
70
- dtype=self.model.dtype,
71
  device=self.model.device
72
  ))
73
 
 
67
  # embed bos prompt
68
  bos_embeds = self.get_input_embeddings()(torch.tensor(
69
  self.config.bos_id,
 
70
  device=self.model.device
71
  ))
72