yutingg
/

custom-distill-bert-for-essay-clarity

yutingg commited on Dec 13, 2023

Commit

660ae5e

•

1 Parent(s): e5cab73

Update to relative import for auto config

Files changed (1) hide show

modeling_essay_clarity.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import transformers
 from torch.utils.data import Dataset, DataLoader
 from transformers import DistilBertModel, DistilBertTokenizer, PreTrainedModel
-from configuration_essay_clarity import DistillBERTClassClarityConfig
 class DistillBERTClassClarity(PreTrainedModel):

 import transformers
 from torch.utils.data import Dataset, DataLoader
 from transformers import DistilBertModel, DistilBertTokenizer, PreTrainedModel
+from .configuration_essay_clarity import DistillBERTClassClarityConfig
 class DistillBERTClassClarity(PreTrainedModel):