Motahar commited on
Commit
4427b5b
1 Parent(s): 4d07349

Updated ganbert.py

Browse files
Files changed (1) hide show
  1. ganbert.py +1 -0
ganbert.py CHANGED
@@ -34,6 +34,7 @@ from datasets import Dataset
34
  import torch.nn as nn
35
  import torch.nn.functional as F
36
  import sys
 
37
 
38
  class GAN(PreTrainedModel):
39
  config_class = GanBertConfig
 
34
  import torch.nn as nn
35
  import torch.nn.functional as F
36
  import sys
37
+ from typing import List, Optional, Tuple, Union
38
 
39
  class GAN(PreTrainedModel):
40
  config_class = GanBertConfig