Fix `NameError: name 'List' is not defined.`

#55

Command:
optimum-cli export openvino ./minicpm-v-2_6/pytorch/ov/fp16 --framework pt --task image-text-to-text --weight-format fp16 --trust-remote-code --model ./minicpm-v-2_6/pytorch

Output:

Traceback (most recent call last):
File "user/venv/bin/optimum-cli", line 8, in <module>
sys.exit(main())
File "user/venv/lib/python3.10/site-packages/optimum/commands/optimum_cli.py", line 208, in main
service.run()
File "user/venv/lib/python3.10/site-packages/optimum/commands/export/openvino.py", line 500, in run
main_export(
File "user/venv/lib/python3.10/site-packages/optimum/exporters/openvino/__main__.py", line 383, in main_export
model = TasksManager.get_model_from_task(
File "user/venv/lib/python3.10/site-packages/optimum/exporters/tasks.py", line 2317, in get_model_from_task
model = model_class.from_pretrained(model_name_or_path, **kwargs)
File "user/venv/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 553, in from_pretrained
model_class = get_class_from_dynamic_module(
File "user/venv/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 553, in get_class_from_dynamic_module
return get_class_in_module(class_name, final_module, force_reload=force_download)
File "user/venv/lib/python3.10/site-packages/transformers/dynamic_module_utils.py", line 250, in get_class_in_module
module_spec.loader.exec_module(module)
File "<frozen importlib._bootstrap_external>", line 883, in exec_module
File "<frozen importlib._bootstrap>", line 241, in _call_with_frames_removed
File "/home/user/.cache/huggingface/modules/transformers_modules/pytorch/modeling_minicpmv.py", line 14, in <module>
from .resampler import Resampler
File "/home/user/.cache/huggingface/modules/transformers_modules/pytorch/resampler.py", line 682, in <module>
) -> List[Tensor]:
NameError: name 'List' is not defined.

Hi,

@finalf0 @yuzaa , could someone from the team please review and merge this PR?

Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment