michaelfeil commited on
Commit
cea4c12
·
1 Parent(s): ce3ff4c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -37,6 +37,9 @@ model = TranslatorCT2fromHfHub(
37
  )
38
  outputs = model.generate(
39
  text=["def print_hello_world():", "def hello_name(name:"],
 
 
 
40
  )
41
  print(outputs)
42
  ```
 
37
  )
38
  outputs = model.generate(
39
  text=["def print_hello_world():", "def hello_name(name:"],
40
+ decode_tok_kwargs=dict(skip_special_tokens=True),
41
+ max_decoding_length=64,
42
+ end_token=["def"]
43
  )
44
  print(outputs)
45
  ```