Updated ganbert.py
Browse files- 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
|