Sarikaa-Sridhar commited on
Commit
cd10160
·
verified ·
1 Parent(s): a8a9d46

Update modeling_yield.py

Browse files
Files changed (1) hide show
  1. modeling_yield.py +2 -2
modeling_yield.py CHANGED
@@ -10,8 +10,8 @@ from transformers import PreTrainedModel
10
  from transformers.modeling_outputs import ModelOutput
11
  from dataclasses import dataclass
12
 
13
- from configuration_yield import YieldConfig
14
- from yield_transformer import YieldTransformer
15
 
16
 
17
  @dataclass
 
10
  from transformers.modeling_outputs import ModelOutput
11
  from dataclasses import dataclass
12
 
13
+ from .configuration_yield import YieldConfig
14
+ from .yield_transformer import YieldTransformer
15
 
16
 
17
  @dataclass