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:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user