app.py Failing for Python 3.9

[john-hawkins]

I successfully installed Donut using a Python3.9 virtual environment.

/opt/homebrew/bin/python3.9 -m venv pyenv3.9 
source pyenv3.9/bin/activate  
pip install .

I did need to install gradio separately because it is not listed in the dependencies of the setup.py file

However when I try and run the demo app it generates an error

% python app.py
config.json: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 405/405 [00:00<00:00, 147kB/s]
pytorch_model.bin: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1.01G/1.01G [00:50<00:00, 20.0MB/s]
tokenizer_config.json: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 535/535 [00:00<00:00, 366kB/s]
sentencepiece.bpe.model: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1.30M/1.30M [00:00<00:00, 1.34MB/s]
added_tokens.json: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 229/229 [00:00<00:00, 338kB/s]
special_tokens_map.json: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 478/478 [00:00<00:00, 472kB/s]
tokenizer.json: 4.01MB [00:02, 1.36MB/s]
Traceback (most recent call last):
File "/Users/john/Projects/donut/app.py", line 49, in
pretrained_model = DonutModel.from_pretrained(args.pretrained_path)
File "/Users/john/Projects/donut/donut/model.py", line 597, in from_pretrained
model = super(DonutModel, cls).from_pretrained(pretrained_model_name_or_path, revision="official", *model_args, **kwargs)
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/transformers/modeling_utils.py", line 277, in _wrapper
return func(*args, **kwargs)
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/transformers/modeling_utils.py", line 5048, in from_pretrained
) = cls._load_pretrained_model(
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/transformers/modeling_utils.py", line 5468, in _load_pretrained_model
_error_msgs, disk_offload_index = load_shard_file(args)
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/transformers/modeling_utils.py", line 843, in load_shard_file
disk_offload_index = _load_state_dict_into_meta_model(
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context
return func(*args, **kwargs)
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/transformers/modeling_utils.py", line 770, in _load_state_dict_into_meta_model
_load_parameter_into_model(model, param_name, param.to(param_device))
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/transformers/modeling_utils.py", line 667, in _load_parameter_into_model
module.load_state_dict({param_type: tensor}, strict=False, assign=True)
File "/Users/john/Projects/donut/pyenv3.9/lib/python3.9/site-packages/torch/nn/modules/module.py", line 2624, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for Linear:
size mismatch for weight: copying a param with shape torch.Size([512, 1024]) from checkpoint, the shape in current model is torch.Size([256, 512]).
model.safetensors: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 803M/803M [00:48<00:00, 16.6MB/s]