Commit
·
1f0431c
1
Parent(s):
0f46607
testin the trick
Browse files- modeling_stacked.py +1 -0
modeling_stacked.py
CHANGED
|
@@ -54,6 +54,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
|
|
| 54 |
# Load floret model
|
| 55 |
self.dummy_param = nn.Parameter(torch.zeros(1))
|
| 56 |
self.safe_floret = SafeFloretWrapper(self.config.filename)
|
|
|
|
| 57 |
# self.model_floret = SafeFloretWrapper(model_floret)
|
| 58 |
# input_ids = "this is a text"
|
| 59 |
|
|
|
|
| 54 |
# Load floret model
|
| 55 |
self.dummy_param = nn.Parameter(torch.zeros(1))
|
| 56 |
self.safe_floret = SafeFloretWrapper(self.config.filename)
|
| 57 |
+
print(self.config)
|
| 58 |
# self.model_floret = SafeFloretWrapper(model_floret)
|
| 59 |
# input_ids = "this is a text"
|
| 60 |
|