Skip to content

Commit bda113d

Browse files
authored
Import transformers if needed pytorch_launcher.py
1 parent 0161a4c commit bda113d

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tools/accuracy_checker/accuracy_checker/launcher/pytorch_launcher.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,7 @@
2020
import importlib
2121
import urllib
2222
import re
23-
import transformers
2423
from collections import OrderedDict
25-
from transformers import AutoConfig
2624
import numpy as np
2725
from ..config import PathField, StringField, DictField, NumberField, ListField, BoolField
2826
from .launcher import Launcher
@@ -179,7 +177,7 @@ def load_module(self, model_cls, module_args, module_kwargs, checkpoint=None, st
179177
return self.prepare_module(module, model_cls)
180178

181179
def load_tranformers_module(self, pretrained_name):
182-
180+
import transformers # pylint: disable=C0415
183181
model_class = getattr(transformers, self.tranformers_class)
184182
module = model_class.from_pretrained(pretrained_name)
185183

0 commit comments

Comments
 (0)