{"id":13511262,"url":"https://github.com/OpenNMT/OpenNMT-py","last_synced_at":"2025-03-30T20:32:53.385Z","repository":{"id":37848259,"uuid":"82841862","full_name":"OpenNMT/OpenNMT-py","owner":"OpenNMT","description":"Open Source Neural Machine Translation and (Large) Language Models in PyTorch","archived":false,"fork":false,"pushed_at":"2024-04-08T14:17:56.000Z","size":321706,"stargazers_count":6547,"open_issues_count":33,"forks_count":2229,"subscribers_count":175,"default_branch":"master","last_synced_at":"2024-04-14T03:59:08.897Z","etag":null,"topics":["deep-learning","language-model","llms","machine-translation","neural-machine-translation","pytorch"],"latest_commit_sha":null,"homepage":"https://opennmt.net/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/OpenNMT.png","metadata":{"files":{"readme":"README.md","changelog":"CHANGELOG.md","contributing":"CONTRIBUTING.md","funding":null,"license":"LICENSE.md","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":"SECURITY.md","support":null,"governance":null,"roadmap":null,"authors":null,"dei":null}},"created_at":"2017-02-22T19:01:50.000Z","updated_at":"2024-04-19T06:41:09.290Z","dependencies_parsed_at":"2022-07-14T23:30:47.033Z","dependency_job_id":"e2a5c9d7-74f6-4993-996d-446ac53788c4","html_url":"https://github.com/OpenNMT/OpenNMT-py","commit_stats":{"total_commits":2276,"total_committers":204,"mean_commits":"11.156862745098039","dds":0.8536906854130053,"last_synced_commit":"aa06c4c202acb653a977c09772d3fb9069824c1b"},"previous_names":[],"tags_count":50,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/OpenNMT%2FOpenNMT-py","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/OpenNMT%2FOpenNMT-py/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/OpenNMT%2FOpenNMT-py/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/OpenNMT%2FOpenNMT-py/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/OpenNMT","download_url":"https://codeload.github.com/OpenNMT/OpenNMT-py/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":222576174,"owners_count":17005447,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["deep-learning","language-model","llms","machine-translation","neural-machine-translation","pytorch"],"created_at":"2024-08-01T03:00:45.595Z","updated_at":"2025-03-30T20:32:53.364Z","avatar_url":"https://github.com/OpenNMT.png","language":"Python","funding_links":[],"categories":["Frameworks","Python","Implementations \u0026 How to reproduce paper's result","Pytorch \u0026 related libraries｜Pytorch \u0026 相关库","Natural Language Processing","Pytorch \u0026 related libraries","Implementations \u0026 How to reproduce paper's result?","自然语言处理 (NLP)","A01_文本生成_文本对话","NLP\u0026PyTorch实战","文本数据和NLP","Developer Resources"],"sub_categories":["Complex, performance-reproducable implementations","NLP \u0026 Speech Processing｜自然语言处理 \u0026 语音处理:","Conversation \u0026 Translation","NLP \u0026 Speech Processing:","其他_文本生成_文本对话","Translation"],"readme":"# Announcement: OpenNMT-py is no longer actively supported.\n\nWe started a new project [Eole](https://eole-nlp.github.io/eole/) available on [Github](https://github.com/eole-nlp/eole)\n\nIt is a spin-off of OpenNMT-py in terms of features but we revamped a lot of stuff.\n\nEole handles NMT, LLM, Encoders as well as a new concept of Estimator within a NMT Model See this [post](https://medium.com/p/05b00b271a47) and this [news](https://www.linkedin.com/posts/vincentnguyenngoc_embarrassingly-small-english-to-german-model-activity-7203400634727841792-FCre?utm_source=share\u0026utm_medium=member_desktop)\n\nIf you are a developer, switch now. If you are a user only, then we will publish the first py-pi versions shortly.\n\n\n# OpenNMT-py: Open-Source Neural Machine Translation and (Large) Language Models\n\n[![Build Status](https://github.com/OpenNMT/OpenNMT-py/workflows/Lint%20\u0026%20Tests/badge.svg)](https://github.com/OpenNMT/OpenNMT-py/actions)\n[![Documentation](https://img.shields.io/badge/docs-latest-blue.svg)](https://opennmt.net/OpenNMT-py/)\n[![Gitter](https://badges.gitter.im/OpenNMT/OpenNMT-py.svg)](https://gitter.im/OpenNMT/OpenNMT-py?utm_source=badge\u0026utm_medium=badge\u0026utm_campaign=pr-badge)\n[![Forum](https://img.shields.io/discourse/status?server=https%3A%2F%2Fforum.opennmt.net%2F)](https://forum.opennmt.net/)\n\nOpenNMT-py is the [PyTorch](https://github.com/pytorch/pytorch) version of the [OpenNMT](https://opennmt.net) project, an open-source (MIT) neural machine translation (and beyond!) framework. It is designed to be research friendly to try out new ideas in translation, language modeling, summarization, and many other NLP tasks. Some companies have proven the code to be production ready.\n\nWe love contributions! Please look at issues marked with the [contributions welcome](https://github.com/OpenNMT/OpenNMT-py/issues?q=is%3Aissue+is%3Aopen+label%3A%22contributions+welcome%22) tag.\n\nBefore raising an issue, make sure you read the requirements and the [Full Documentation](https://opennmt.net/OpenNMT-py/) examples.\n\nUnless there is a bug, please use the [Forum](https://forum.opennmt.net) or [Gitter](https://gitter.im/OpenNMT/OpenNMT-py) to ask questions.\n\n----\n## For beginners:\n\nThere is a step-by-step and explained tuto (Thanks to Yasmin Moslem): [Tutorial](https://github.com/ymoslem/OpenNMT-Tutorial)\n\nPlease try to read and/or follow before raising newbies issues.\n\nOtherwise you can just have a look at the [Quickstart](https://opennmt.net/OpenNMT-py/quickstart.html) steps\n\n----\n## New:\n\n* You will need Pytorch v2 preferably v2.2 which fixes some `scaled_dot_product_attention` issues\n* LLM support with converters for: Llama (+ Mistral), OpenLlama, Redpajama, MPT-7B, Falcon.\n* Support for 8bit and 4bit quantization along with LoRA adapters, with or without checkpointing.\n* You can finetune 7B and 13B models on a single RTX 24GB with 4-bit quantization.\n* Inference can be forced in 4/8bit using the same layer quantization as in finetuning.\n* Tensor parallelism when the model does not fit on one GPU's memory (both training and inference)\n* Once your model is finetuned you can run inference either with OpenNMT-py or faster with CTranslate2.\n* MMLU evaluation script, see results [here](https://github.com/OpenNMT/OpenNMT-py/blob/master/eval_llm/MMLU/readme.md)\n\nFor all usecases including NMT, you can now use Multiquery instead of Multihead attention (faster at training and inference) and remove biases from all Linear (QKV as well as FeedForward modules).\n\n\nIf you used previous versions of OpenNMT-py, you can check the [Changelog](https://github.com/OpenNMT/OpenNMT-py/blob/master/CHANGELOG.md) or the [Breaking Changes](https://github.com/OpenNMT/OpenNMT-py/blob/master/docs/source/changes.md)\n\n----\n\n## Tutorials:\n\n* How to replicate Vicuna with a 7B or 13B llama (or Open llama, MPT-7B, Redpajama)  Language Model: [Tuto Vicuna](https://github.com/OpenNMT/OpenNMT-py/blob/master/docs/source/examples/replicate_vicuna/ReplicateVicuna.md)\n* How to finetune NLLB-200 with your dataset: [Tuto Finetune NLLB-200](https://forum.opennmt.net/t/finetuning-and-curating-nllb-200-with-opennmt-py/5238)\n* How to create a simple OpenNMT-py REST Server: [Tuto REST](https://forum.opennmt.net/t/simple-opennmt-py-rest-server/1392)\n* How to create a simple Web Interface: [Tuto Streamlit](https://forum.opennmt.net/t/simple-web-interface/4527)\n* Replicate the WMT17 en-de experiment: [WMT17 ENDE](https://github.com/OpenNMT/OpenNMT-py/blob/master/docs/source/examples/wmt17/Translation.md)\n\n----\n\n## Setup\n\n### Using docker\n\nTo facilitate setup and reproducibility, some docker images are made available via the Github Container Registry:\nhttps://github.com/OpenNMT/OpenNMT-py/pkgs/container/opennmt-py\n\nYou can adapt the workflow and build your own image(s) depending on specific needs by using `build.sh` and `Dockerfile` in the `docker` directory of the repo.\n\n```\ndocker pull ghcr.io/opennmt/opennmt-py:3.4.3-ubuntu22.04-cuda12.1\n```\n\nExample oneliner to run a container and open a bash shell within it\n```\ndocker run --rm -it --runtime=nvidia ghcr.io/opennmt/opennmt-py:test-ubuntu22.04-cuda12.1\n```\nNote: you need to have the [Nvidia Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html) (formerly nvidia-docker) installed to properly take advantage of the CUDA/GPU features.\n\nDepending on your needs you can add various flags:\n- `-p 5000:5000` to forward some exposed port from your container to your host;\n- `-v /some/local/directory:/some/container/directory` to mount some local directory to some container directory;\n- `--entrypoint some_command` to directly run some specific command as the container entry point (instead of the default bash shell);\n\n### Installing locally\n\nOpenNMT-py requires:\n\n- Python \u003e= 3.8\n- PyTorch \u003e= 2.0 \u003c2.2\n\nInstall `OpenNMT-py` from `pip`:\n```bash\npip install OpenNMT-py\n```\n\nor from the source:\n```bash\ngit clone https://github.com/OpenNMT/OpenNMT-py.git\ncd OpenNMT-py\npip install -e .\n```\n\nNote: if you encounter a `MemoryError` during installation, try to use `pip` with `--no-cache-dir`.\n\n*(Optional)* Some advanced features (e.g. working pretrained models or specific transforms) require extra packages, you can install them with:\n\n```bash\npip install -r requirements.opt.txt\n```\n\n### Manual installation of some dependencies\n\nApex is highly recommended to have fast performance (especially the legacy fusedadam optimizer and FusedRMSNorm)\n\n```shell\ngit clone https://github.com/NVIDIA/apex\ncd apex\npip3 install -v --no-build-isolation --config-settings --build-option=\"--cpp_ext --cuda_ext --deprecated_fused_adam --xentropy --fast_multihead_attn\" ./\ncd ..\n```\n\nFlash attention:\n\nAs of Oct. 2023 flash attention 1 has been upstreamed to pytorch v2 but it is recommended to use flash attention 2 with v2.3.1 for sliding window attention support.\n\nWhen using regular `position_encoding=True` or Rotary with `max_relative_positions=-1` OpenNMT-py will try to use an optimized dot-product path.\n\nif you want to use [flash attention](https://github.com/Dao-AILab/flash-attention#installation-and-features) then you need to manually install it first:\n\n```bash\npip install flash-attn --no-build-isolation\n```\n\nif flash attention 2 is not installed, then we will use `F.scaled_dot_product_attention` from pytorch 2.x\n\nWhen using `max_relative_positions \u003e 0` or Alibi `max_relative_positions=-2` OpenNMT-py will use its legacy code for matrix multiplications.\n\nflash attention and `F.scaled_dot_product_attention` are a bit faster and saves some GPU memory.\n\n\nAWQ:\n\nIf you want to run inference or quantize an AWQ model you will need AutoAWQ.\n\nFor [AutoAWQ](https://github.com/casper-hansen/AutoAWQ):\n    pip install autoawq\n\n\n## Documentation \u0026 FAQs\n\n[Full HTML Documentation](https://opennmt.net/OpenNMT-py/quickstart.html)\n\n[FAQs](https://github.com/OpenNMT/OpenNMT-py/blob/master/docs/source/FAQ.md)\n\n## Acknowledgements\n\nOpenNMT-py is run as a collaborative open-source project.\nProject was incubated by Systran and Harvard NLP in 2016 in Lua and ported to Pytorch in 2017.\n\nCurrent maintainers (since 2018):\n\n[François Hernandez](https://github.com/francoishernandez)\n[Vincent Nguyen](https://github.com/vince62s) (Seedfall)\n\n## Citation\n\nIf you are using OpenNMT-py for academic work, please cite the initial [system demonstration paper](https://www.aclweb.org/anthology/P17-4012) published in ACL 2017:\n\n```\n@misc{klein2018opennmt,\n      title={OpenNMT: Neural Machine Translation Toolkit}, \n      author={Guillaume Klein and Yoon Kim and Yuntian Deng and Vincent Nguyen and Jean Senellart and Alexander M. Rush},\n      year={2018},\n      eprint={1805.11462},\n      archivePrefix={arXiv},\n      primaryClass={cs.CL}\n}\n```\n\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2FOpenNMT%2FOpenNMT-py","html_url":"https://awesome.ecosyste.ms/projects/github.com%2FOpenNMT%2FOpenNMT-py","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2FOpenNMT%2FOpenNMT-py/lists"}