v1.1

+ fixed README.md + fixed requirements
Xeadriel · Dec 7, 2022 · 517f2a5 · 517f2a5
1 parent c6c67c5
commit 517f2a5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -87,8 +87,8 @@ The models will be downloaded to `checkpoints/` and decompressed.
 
 The saved models can be reloaded by
 ```python
-import torch
-model = torch.load('checkpoints/wacky_mlplanardiv_d2_l4_k10')
+import firelang
+model = firelang.FireWord.from_pretrained('checkpoints/v1.1/wacky_mlplanardiv_d2_l4_k10')
 ```
 
 ## 2. Run benchmarks

diff --git a/requirements-cu10.txt b/requirements-cu10.txt
@@ -1,5 +1,5 @@
 -f https://download.pytorch.org/whl/cu102/torch_stable.html
-corpusit==0.1.2
+corpusit==0.1.3
 faster_fifo<=1.2.0
 nltk<=3.7
 numpy==1.20

diff --git a/requirements-cu11.txt b/requirements-cu11.txt
@@ -1,5 +1,5 @@
 -f https://download.pytorch.org/whl/cu111/torch_stable.html
-corpusit==0.1.2
+corpusit==0.1.3
 faster_fifo<=1.2.0
 nltk<=3.7
 numpy==1.20