ModuleNotFoundError: No module named 'mm_builder'

[bozagina]

[bingwork]

hi, @bozagina
I noticed that in the code you referenced from the image, starting from line 3 in inference.py, I saw the line from mm_builder import load_pretrained_model.
However, in the version of the code in the official repository, this line is not included.
Could you please clarify if there has been an update or if there might be a discrepancy between the versions?

[bozagina]

hi, @bozagina I noticed that in the code you referenced from the image, starting from line 3 in inference.py, I saw the line from mm_builder import load_pretrained_model. However, in the version of the code in the official repository, this line is not included. Could you please clarify if there has been an update or if there might be a discrepancy between the versions?

hi, i was using the inference.py script as shown in the attached image：

Additionally, when I tried to use the model weights that have already been downloaded, I encountered the following errors:

[bingwork]

Ah, I got it. Please update to use the inference.py from MMAlaya GitHub repository, as I checked and it still works now. I'm sorry, but the inference.py on HuggingFace is incorrect. Also, please make sure to use transformers==4.33.0.

[bozagina]

Ah, I got it. Please update to use the inference.py from MMAlaya GitHub repository, as I checked and it still works now. I'm sorry, but the inference.py on HuggingFace is incorrect. Also, please make sure to use transformers==4.33.0.
"When I use the command CUDA_VISIBLE_DEVICES=0 python infer_mmalaya.py, it works fine.

when I use the command "CUDA_VISIBLE_DEVICES=0 python infer_mmalaya.py --model_path "/mnt/znzz/william/inference/model_zoo/DataCanvas/MMAlaya""

with the pre-downloaded model weights, as shown below:"

[bingwork]

It seems like the attention.py file is missing in your /root/.cache/huggingface/modules/transformers_modules/MMAlaya directory. You could try running the following copy command to fix the issue:

cp $(ls | grep -vE 'pytorch_model-00001-of-00002.bin|pytorch_model-00002-of-00002.bin') /root/.cache/huggingface/modules/transformers_modules/MMAlaya