Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,7 @@ torch.cuda.empty_cache()
|
|
| 11 |
|
| 12 |
model_name = "hfl/chinese-roberta-wwm-ext"
|
| 13 |
|
| 14 |
-
class
|
| 15 |
def __init__(self, config, model_name_or_path):
|
| 16 |
super().__init__(config)
|
| 17 |
# Load backbone with pretrained weights if desired
|
|
|
|
| 11 |
|
| 12 |
model_name = "hfl/chinese-roberta-wwm-ext"
|
| 13 |
|
| 14 |
+
class MultiTaskRoBert(BertPreTrainedModel):
|
| 15 |
def __init__(self, config, model_name_or_path):
|
| 16 |
super().__init__(config)
|
| 17 |
# Load backbone with pretrained weights if desired
|