justinpinkney
/

falcon-7b

Text Generation

RefinedWebModel

text-generation-inference

Model card Files Files and versions Community

justinpinkney commited on Jul 7, 2023

Commit

d532182

·

1 Parent(s): 1b7419e

fix indent

Files changed (1) hide show

modelling_RW.py +1 -1

modelling_RW.py CHANGED Viewed

@@ -696,7 +696,7 @@ class RWForCausalLM(RWPreTrainedModel):
     def set_output_embeddings(self, new_embeddings: torch.Tensor):
         self.lm_head = new_embeddings
-        def prepare_inputs_for_generation(
         self,
         input_ids: torch.LongTensor,
         past: Optional[torch.Tensor] = None,

     def set_output_embeddings(self, new_embeddings: torch.Tensor):
         self.lm_head = new_embeddings
+    def prepare_inputs_for_generation(
         self,
         input_ids: torch.LongTensor,
         past: Optional[torch.Tensor] = None,