sebastian-hofstaetter
/

prettr-distilbert-split_at_3-margin_mse-T2-msmarco

knowledge-distillation

Model card Files Files and versions

sebastian-hofstaetter commited on Jul 10, 2021

Commit

b872cbc

·

1 Parent(s): 04f9b13

fix imports

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ For more information, training data, source code, and a minimal usage example pl
 ## Model Code
 ````python
-from transformers import *
 from transformers.models.distilbert.modeling_distilbert import *
 import math
 import torch

 ## Model Code
 ````python
+from transformers import DistilBertModel,AutoTokenizer
 from transformers.models.distilbert.modeling_distilbert import *
 import math
 import torch