Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
amaiya committed Mar 10, 2021
1 parent 1a6498c commit 1f3768a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ Note that, `transformers>=4.0.0` included a complete reogranization of the modul
For instance, suppose you trained a DistilBERT model and saved the resultant predictor using an older version of **ktrain** with: `predictor.save('/tmp/my_predictor/')`. After upgrading to the newest version of **ktrain**, you will find that `ktrain.load_predictor('/tmp/my_predictor`) will throw an error unless you follow one of the two approaches below:

**Approach 1: Manually edit `tf_model.preproc` file:**
Open `tf_model.preproc` with an editor like **vim** and edit it to replace old module locations with new module locations (changes below show example changes for a DistilBERT model):
Open `tf_model.preproc` with an editor like **vim** and edit it to replace old module locations with new module locations (example changes for a DistilBERT model shown below):
```python
# change transformers.configuration_distilbert to transformers.models.distilbert.configuration_distilbert
# change transformers.modeling_tf_auto to transformers.models.auto.modeling_tf_auto
Expand Down

0 comments on commit 1f3768a

Please sign in to comment.