emanuelaboros commited on
Commit
775d548
1 Parent(s): e74df63

Upload model

Browse files
Files changed (1) hide show
  1. models.py +1 -1
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