Delete modeling.py
Browse files- modeling.py +0 -18
modeling.py
DELETED
@@ -1,18 +0,0 @@
|
|
1 |
-
from transformers import AutoConfig
|
2 |
-
from .modeling_custom import CustomModel
|
3 |
-
|
4 |
-
def from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs):
|
5 |
-
# Load the configuration
|
6 |
-
config = AutoConfig.from_pretrained(pretrained_model_name_or_path)
|
7 |
-
|
8 |
-
# Fetch the number of emotion labels from the config
|
9 |
-
num_emotion_labels = config.num_emotion_labels
|
10 |
-
|
11 |
-
# Load the model with the correct configuration
|
12 |
-
model = CustomModel.from_pretrained(
|
13 |
-
pretrained_model_name_or_path,
|
14 |
-
num_emotion_labels=num_emotion_labels,
|
15 |
-
*model_args,
|
16 |
-
**kwargs
|
17 |
-
)
|
18 |
-
return model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|