Subi003 commited on
Commit
4f6c161
·
verified ·
1 Parent(s): 4b8c421

Update modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +5 -5
modeling.py CHANGED
@@ -41,11 +41,11 @@ class RobertaEmoPillars(PreTrainedModel):
41
  self.post_init() # ensures HF weights init
42
 
43
  def forward(self, input_ids=None, attention_mask=None, **kwargs):
44
- inputs = {
45
- "input_ids": input_ids,
46
- "attention_mask": attention_mask,
47
- **kwargs
48
- }
49
 
50
 
51
  # --- Registration for AutoModel ---
 
41
  self.post_init() # ensures HF weights init
42
 
43
  def forward(self, input_ids=None, attention_mask=None, **kwargs):
44
+ inputs = {
45
+ "input_ids": input_ids,
46
+ "attention_mask": attention_mask,
47
+ **kwargs
48
+ }
49
 
50
 
51
  # --- Registration for AutoModel ---