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
|