Upload modeling_normwear.py
Browse files- modeling_normwear.py +6 -0
modeling_normwear.py
CHANGED
|
@@ -37,6 +37,12 @@ class NormWear2Model(PreTrainedModel):
|
|
| 37 |
|
| 38 |
self.post_init()
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
def forward(self, *args, **kwargs):
|
| 41 |
return self.normwear(*args, **kwargs)
|
| 42 |
|
|
|
|
| 37 |
|
| 38 |
self.post_init()
|
| 39 |
|
| 40 |
+
def forward_encoder(self, *args, **kwargs):
|
| 41 |
+
return self.normwear.forward_encoder(*args, **kwargs)
|
| 42 |
+
|
| 43 |
+
def forward_decoder(self, *args, **kwargs):
|
| 44 |
+
return self.normwear.forward_decoder(*args, **kwargs)
|
| 45 |
+
|
| 46 |
def forward(self, *args, **kwargs):
|
| 47 |
return self.normwear(*args, **kwargs)
|
| 48 |
|