translate-package 0.0.9__tar.gz → 0.1.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (32) hide show
  1. {translate_package-0.0.9 → translate_package-0.1.1}/PKG-INFO +1 -1
  2. {translate_package-0.0.9 → translate_package-0.1.1}/setup.py +1 -1
  3. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/machine_translation.py +3 -3
  4. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package.egg-info/PKG-INFO +1 -1
  5. {translate_package-0.0.9 → translate_package-0.1.1}/setup.cfg +0 -0
  6. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/__init__.py +0 -0
  7. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/data/__init__.py +0 -0
  8. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/data/__pycache__/__init__.cpython-310.pyc +0 -0
  9. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/data/__pycache__/data_preparation.cpython-310.pyc +0 -0
  10. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/data/data_preparation.py +0 -0
  11. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/errors/__init__.py +0 -0
  12. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/errors/__pycache__/__init__.cpython-310.pyc +0 -0
  13. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/__init__.py +0 -0
  14. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/__pycache__/__init__.cpython-310.pyc +0 -0
  15. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/__pycache__/code_generation.cpython-310.pyc +0 -0
  16. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/__pycache__/machine_translation.cpython-310.pyc +0 -0
  17. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/gradient_observation.py +0 -0
  18. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/lstm.py +0 -0
  19. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/tokenization/__init__.py +0 -0
  20. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/tokenization/__pycache__/__init__.cpython-310.pyc +0 -0
  21. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/tokenization/__pycache__/load_tokenizer.cpython-310.pyc +0 -0
  22. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/tokenization/__pycache__/train_tokenizer.cpython-310.pyc +0 -0
  23. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/tokenization/load_tokenizer.py +0 -0
  24. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/tokenization/train_tokenizer.py +0 -0
  25. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/utils/__init__.py +0 -0
  26. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/utils/__pycache__/__init__.cpython-310.pyc +0 -0
  27. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/utils/__pycache__/checkpoint.cpython-310.pyc +0 -0
  28. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package/utils/checkpoint.py +0 -0
  29. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package.egg-info/SOURCES.txt +0 -0
  30. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package.egg-info/dependency_links.txt +0 -0
  31. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package.egg-info/requires.txt +0 -0
  32. {translate_package-0.0.9 → translate_package-0.1.1}/translate_package.egg-info/top_level.txt +0 -0
{translate_package-0.0.9 → translate_package-0.1.1}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: translate_package
3
- Version: 0.0.9
3
+ Version: 0.1.1
4
4
  Summary: Contain functions and classes to efficiently train a sequence to sequence to translate between two languages.
5
5
  Author: Oumar Kane
6
6
  Author-email: oumar.kane@univ-thies.sn
{translate_package-0.0.9 → translate_package-0.1.1}/setup.py RENAMED
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
2
2
 
3
3
  setup(
4
4
  name="translate_package",
5
- version="0.0.9",
5
+ version="0.1.1",
6
6
  author="Oumar Kane",
7
7
  packages=find_packages(),
8
8
  author_email="oumar.kane@univ-thies.sn",
{translate_package-0.0.9 → translate_package-0.1.1}/translate_package/models/machine_translation.py RENAMED
@@ -158,7 +158,7 @@ class MachineTranslationTransformer(pl.LightningModule):
158
158
  warmup_init = False
159
159
  )
160
160
 
161
- elif self.model_generation in ["bart", "mbart"]:
161
+ elif self.model_generation in ["bart", "mbart", "nllb"]:
162
162
 
163
163
  optimizer = torch.optim.AdamW(
164
164
  self.parameters(), lr=self.lr, weight_decay=self.weight_decay
@@ -170,11 +170,11 @@ class MachineTranslationTransformer(pl.LightningModule):
170
170
  self.parameters(), lr=self.lr, weight_decay=self.weight_decay
171
171
  )
172
172
 
173
- if self.model_generation in ["t5", "lstm", "mt5"]:
173
+ if self.model_generation in ["t5", "lstm"]:
174
174
 
175
175
  return [optimizer]
176
176
 
177
- elif self.model_generation in ["bart"]:
177
+ elif self.model_generation in ["bart", "mt5", "nllb"]:
178
178
 
179
179
  scheduler = get_linear_schedule_with_warmup(
180
180
  optimizer,
{translate_package-0.0.9 → translate_package-0.1.1}/translate_package.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: translate-package
3
- Version: 0.0.9
3
+ Version: 0.1.1
4
4
  Summary: Contain functions and classes to efficiently train a sequence to sequence to translate between two languages.
5
5
  Author: Oumar Kane
6
6
  Author-email: oumar.kane@univ-thies.sn