mjwong commited on
Commit
bf56f64
·
verified ·
1 Parent(s): 329e8fa

Delete modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +0 -13
modeling.py DELETED
@@ -1,13 +0,0 @@
1
- from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
-
3
- class CustomModel:
4
- @classmethod
5
- def from_pretrained(cls, model_name):
6
- model = AutoModelForSequenceClassification.from_pretrained(model_name)
7
- tokenizer = AutoTokenizer.from_pretrained(model_name)
8
-
9
- # Force setting pad_token in the tokenizer
10
- if tokenizer.pad_token is None:
11
- tokenizer.pad_token = tokenizer.eos_token
12
-
13
- return model, tokenizer