ad17171717 / youtube-tutorials Goto Github PK

downloading nougat checkpoint version 0.1.0-small to path /root/.cache/torch/hub/nougat-0.1.0-small
config.json: 100% 557/557 [00:00<00:00, 3.39Mb/s]
pytorch_model.bin: 100% 956M/956M [00:04<00:00, 201Mb/s]
special_tokens_map.json: 100% 96.0/96.0 [00:00<00:00, 531kb/s]
tokenizer.json: 100% 2.04M/2.04M [00:00<00:00, 9.18Mb/s]
tokenizer_config.json: 100% 106/106 [00:00<00:00, 521kb/s]
INFO:root:Output directory does not exist. Creating output directory.
/usr/local/lib/python3.10/dist-packages/torch/functional.py:507: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3549.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
0% 0/1 [00:00<?, ?it/s]
Traceback (most recent call last):
File "/usr/local/bin/nougat", line 8, in
sys.exit(main())
File "/usr/local/lib/python3.10/dist-packages/predict.py", line 167, in main
model_output = model.inference(
File "/usr/local/lib/python3.10/dist-packages/nougat/model.py", line 592, in inference
decoder_output = self.decoder.model.generate(
File "/usr/local/lib/python3.10/dist-packages/torch/utils/_contextlib.py", line 115, in decorate_context
return func(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/transformers/generation/utils.py", line 1576, in generate
result = self._greedy_search(
File "/usr/local/lib/python3.10/dist-packages/transformers/generation/utils.py", line 2491, in _greedy_search
model_inputs = self.prepare_inputs_for_generation(input_ids, **model_kwargs)
TypeError: BARTDecoder.prepare_inputs_for_inference() got an unexpected keyword argument 'cache_position'
-> Cannot close object, library is destroyed. This may cause a memory leak!