import torch
import torch.nn as nn
import transformers


from models.new_encoder import NewTransformer as NewEncoder

import torch
import torch.nn as nn
import transformers


from models.new_encoder import NewTransformer as NewEncoder
