emanuelaboros
commited on
Commit
•
775d548
1
Parent(s):
e74df63
Upload model
Browse files
models.py
CHANGED
@@ -4,8 +4,8 @@ import torch.nn as nn
|
|
4 |
from transformers import PreTrainedModel, AutoModel, AutoConfig
|
5 |
from torch.nn import CrossEntropyLoss
|
6 |
from typing import Optional, Tuple, Union
|
7 |
-
from configuration_extended_multitask import ImpressoConfig
|
8 |
import logging
|
|
|
9 |
|
10 |
logger = logging.getLogger(__name__)
|
11 |
|
|
|
4 |
from transformers import PreTrainedModel, AutoModel, AutoConfig
|
5 |
from torch.nn import CrossEntropyLoss
|
6 |
from typing import Optional, Tuple, Union
|
|
|
7 |
import logging
|
8 |
+
from configuration_extended_multitask import ImpressoConfig
|
9 |
|
10 |
logger = logging.getLogger(__name__)
|
11 |
|