update

Author: Nghi Bui

codetf/models/causal_lm_models/__init__.py

@@ -87,7 +87,6 @@ def forward(self, sources, max_length=512):
         generated_ids = self.model.generate(**encoding, 
                                             max_length=max_length)
 
-        print(generated_ids)
         predictions = self.tokenizer.batch_decode(generated_ids, truncate_before_pattern=[r"\n\n^#", "^'''", "\n\n\n"])
         return predictions
 

setup.py

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

test_inference/test_starcoder_nl2code.py

@@ -8,7 +8,7 @@
             model_type="starcoder-15.5B", is_eval=True,
             load_in_8bit=True, weight_sharding=False)
 
-prompts = "# this function prints hello world"
+prompts = "def print_hello_world():"
 code_snippets = model.predict([prompts])
 
 print(code_snippets)