e2k 0.2.0__tar.gz → 0.2.2__tar.gz

{e2k-0.2.0/src/e2k.egg-info → e2k-0.2.2}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.2
 Name: e2k
-Version: 0.2.0
+Version: 0.2.2
 Summary: A tool for automatic English to Katakana conversion
 Author-email: Patchethium <asguftuikh@gmail.com>
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: numpy<2.0,>=1.0
+Requires-Dist: numpy<3.0,>=1.0
 
 # English to Katakana Translator
 
@@ -154,7 +154,7 @@ After installing the dependencies, `torch` will be added as a development depend
 python train.py --data ./vendor/katakana_dict.jsonl
 ```
 
-It takes around 10 minutes on a desktop CPU. The model will be saved as `vendor/model-{p2k/c2k}-e-{epoch}.pth`.
+It takes around 10 minutes on a desktop CPU. The model will be saved as `vendor/model-{p2k/c2k}-e{epoch}.pth`.
 
 Also, you'll need to either download the `katakana_dict.jsonl` from the releases or create it yourself using the `extract.py` script.
 

{e2k-0.2.0 → e2k-0.2.2}/README.md

@@ -145,7 +145,7 @@ After installing the dependencies, `torch` will be added as a development depend
 python train.py --data ./vendor/katakana_dict.jsonl
 ```
 
-It takes around 10 minutes on a desktop CPU. The model will be saved as `vendor/model-{p2k/c2k}-e-{epoch}.pth`.
+It takes around 10 minutes on a desktop CPU. The model will be saved as `vendor/model-{p2k/c2k}-e{epoch}.pth`.
 
 Also, you'll need to either download the `katakana_dict.jsonl` from the releases or create it yourself using the `extract.py` script.
 

{e2k-0.2.0 → e2k-0.2.2}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "e2k"
-version = "0.2.0"
+version = "0.2.2"
 description = "A tool for automatic English to Katakana conversion"
 readme = "README.md"
 authors = [{ name = "Patchethium", email = "asguftuikh@gmail.com" }]
 requires-python = ">=3.10"
-dependencies = ["numpy>=1.0,<2.0"]
+dependencies = ["numpy>=1.0,<3.0"]
 
 [build-system]
 requires = ["setuptools", "wheel"]

{e2k-0.2.0 → e2k-0.2.2}/src/e2k/inference.py

@@ -140,6 +140,11 @@ class MHA:
 
 class S2S:
     def __init__(self, weights: Dict[str, np.ndarray], max_len: int = 16):
+        # fp32 is usually faster than fp16 on CPU
+        new_weight = {}
+        for k, v in weights.items():
+            new_weight[k] = v.astype(np.float32)
+        weights = new_weight
         self.e_emb = Embedding(weights["e_emb.weight"])
         self.k_emb = Embedding(weights["k_emb.weight"])
         self.encoder = GRU(

{e2k-0.2.0 → e2k-0.2.2/src/e2k.egg-info}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.2
 Name: e2k
-Version: 0.2.0
+Version: 0.2.2
 Summary: A tool for automatic English to Katakana conversion
 Author-email: Patchethium <asguftuikh@gmail.com>
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
-Requires-Dist: numpy<2.0,>=1.0
+Requires-Dist: numpy<3.0,>=1.0
 
 # English to Katakana Translator
 
@@ -154,7 +154,7 @@ After installing the dependencies, `torch` will be added as a development depend
 python train.py --data ./vendor/katakana_dict.jsonl
 ```
 
-It takes around 10 minutes on a desktop CPU. The model will be saved as `vendor/model-{p2k/c2k}-e-{epoch}.pth`.
+It takes around 10 minutes on a desktop CPU. The model will be saved as `vendor/model-{p2k/c2k}-e{epoch}.pth`.
 
 Also, you'll need to either download the `katakana_dict.jsonl` from the releases or create it yourself using the `extract.py` script.
 

e2k-0.2.2/src/e2k.egg-info/requires.txt

@@ -0,0 +1 @@
+numpy<3.0,>=1.0

e2k-0.2.0/src/e2k.egg-info/requires.txt

@@ -1 +0,0 @@
-numpy<2.0,>=1.0