Instructions to use chi-vi/hirashiba-mt-tiny-zh-vi with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use chi-vi/hirashiba-mt-tiny-zh-vi with Transformers:
# Use a pipeline as a high-level helper # Warning: Pipeline type "translation" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("translation", model="chi-vi/hirashiba-mt-tiny-zh-vi")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("chi-vi/hirashiba-mt-tiny-zh-vi") model = AutoModelForSeq2SeqLM.from_pretrained("chi-vi/hirashiba-mt-tiny-zh-vi") - Notebooks
- Google Colab
- Kaggle
Add usage example
Browse files
README.md
CHANGED
|
@@ -20,8 +20,31 @@ Hira's intelligence, Shiba's speed
|
|
| 20 |
|
| 21 |
Hirashiba-MT-zh-vi is a model used for gatekeeping and refilling water.
|
| 22 |
|
|
|
|
| 23 |
|
|
|
|
|
|
|
|
|
|
| 24 |
|
|
|
|
|
|
|
|
|
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
|
|
|
| 20 |
|
| 21 |
Hirashiba-MT-zh-vi is a model used for gatekeeping and refilling water.
|
| 22 |
|
| 23 |
+
## Usage
|
| 24 |
|
| 25 |
+
```python
|
| 26 |
+
|
| 27 |
+
from transformers import MarianMTModel, MarianTokenizer
|
| 28 |
|
| 29 |
+
model_name = "chi-vi/hirashiba-mt-tiny-zh-vi"
|
| 30 |
+
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
| 31 |
+
model = MarianMTModel.from_pretrained(model_name)
|
| 32 |
|
| 33 |
+
def translate(lines):
|
| 34 |
+
inputs = tokenizer(lines, return_tensors="pt", padding=True)
|
| 35 |
+
translated = model.generate(**inputs)
|
| 36 |
+
return [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
|
| 37 |
|
| 38 |
+
with open('sample.txt') as f:
|
| 39 |
+
src_text = f.readlines()
|
| 40 |
+
|
| 41 |
+
import time
|
| 42 |
+
|
| 43 |
+
start = time.time()
|
| 44 |
+
translated = translate(src_text)
|
| 45 |
+
end = time.time()
|
| 46 |
+
|
| 47 |
+
print(translated)
|
| 48 |
+
print(end - start)
|
| 49 |
+
```
|
| 50 |
|