added type hints for BART model (#16270)

* added type hints for BART model

* make fixup, adding imports to copied files

* Adding some missing types to cookiecutter

* Adding some missing types to cookiecutter

* Adding some missing types to cookiecutter

Co-authored-by: matt <rocketknight1@gmail.com>
This commit is contained in:
Robot Jelly
2022-03-21 20:48:01 +05:30
committed by GitHub
parent 460f36d352
commit d50f62f2de
19 changed files with 482 additions and 461 deletions

View File

@@ -25,7 +25,7 @@ import math
import os
import sys
from dataclasses import dataclass, field
from typing import Optional
from typing import Optional, List
import datasets
from datasets import load_dataset

View File

@@ -1571,7 +1571,7 @@ class {{cookiecutter.camelcase_modelname}}ForQuestionAnswering({{cookiecutter.ca
import math
import copy
import random
from typing import Optional, Tuple
from typing import Optional, Tuple, List, Union
import torch
from torch import nn