Commit
·
38cd562
1
Parent(s):
1f0431c
testin the trick
Browse files- modeling_stacked.py +1 -0
modeling_stacked.py
CHANGED
|
@@ -55,6 +55,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
|
|
| 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 |
|
|
|
|
| 55 |
self.dummy_param = nn.Parameter(torch.zeros(1))
|
| 56 |
self.safe_floret = SafeFloretWrapper(self.config.filename)
|
| 57 |
print(self.config)
|
| 58 |
+
print(self.config["filename"])
|
| 59 |
# self.model_floret = SafeFloretWrapper(model_floret)
|
| 60 |
# input_ids = "this is a text"
|
| 61 |
|