Commit
·
cea4c12
1
Parent(s):
ce3ff4c
Update README.md
Browse files
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 |
```
|