| language: | |
| - en | |
| - km | |
| license: mit | |
| library_name: netra-nmt | |
| pipeline_tag: translation | |
| tags: | |
| - translation | |
| - khmer | |
| - english | |
| - diffusion | |
| - non-autoregressive | |
| # netra-nmt (diffusion, experimental) | |
| A non-autoregressive **discrete-diffusion** decoder variant of netra-nmt, same | |
| encoder-decoder size (~90M params) as the autoregressive model, trading some | |
| translation quality for fixed-cost, parallel CPU decoding (see the main repo | |
| README for the quality/latency comparison). | |
| ## Usage | |
| ```bash | |
| pip install netra-nmt | |
| ``` | |
| ```python | |
| from netra_nmt import DiffusionTranslator | |
| t = DiffusionTranslator(repo_id="Darayut/netra-nmt-small-diffusion", steps=10) | |
| print(t.translate("Hello, how are you?", direction="en2km")) | |
| print(t.translate("សួស្តី", direction="km2en")) | |
| ``` | |
| Files: `model.safetensors` (fp16 weights), `config.json`, `spm_32k.model` (SentencePiece tokenizer). | |