Commit
·
bbe80a9
1
Parent(s):
145a8f8
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 |
```
|
|
@@ -45,11 +48,6 @@ print(outputs)
|
|
| 45 |
This is just a quantized version. Licence conditions are intended to be idential to original huggingface repo.
|
| 46 |
|
| 47 |
# Original description
|
| 48 |
-
|
| 49 |
-
tags:
|
| 50 |
-
- ctranslate2
|
| 51 |
-
- int8
|
| 52 |
-
- float16
|
| 53 |
|
| 54 |
|
| 55 |
# CodeT5+ 770M (further tuned on Python)
|
|
|
|
| 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 |
```
|
|
|
|
| 48 |
This is just a quantized version. Licence conditions are intended to be idential to original huggingface repo.
|
| 49 |
|
| 50 |
# Original description
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
# CodeT5+ 770M (further tuned on Python)
|