remove beamsize in causal lm

Author: bdqnghi

codetf/models/causal_lm_models/__init__.py

@@ -72,8 +72,7 @@ def forward(self, sources):
         input_ids = encoding.input_ids.to(self.device)
         attention_mask = encoding.attention_mask.to(self.device)
         generated_ids = self.model.generate(input_ids, attention_mask=attention_mask, 
-                                            max_length=self.max_prediction_length,
-                                            num_beams=self.beam_size)
+                                            max_length=self.max_prediction_length)
 
         predictions = self.tokenizer.batch_decode(generated_ids, truncate_before_pattern=[r"\n\n^#", "^'''", "\n\n\n"])
         return predictions

setup.py

@@ -31,7 +31,7 @@
 
 setup(
   name = 'salesforce-codetf',
-  version = "1.0.0",
+  version = "1.0.1",
   py_modules = ['codetf'],
   description = 'CodeTF: A Transformer-based Library for Code Intelligence',
   author = 'Nghi D. Q. Bui',