-
Notifications
You must be signed in to change notification settings - Fork 52
Description
When I'm trying to run the demo, it says the following:
python3 playground.py --stage1_weight ../WEIGHT_FOLDER/stage1_cc3m.ckpt --test_weight ../WEIGHT_FOLDER/stage2_vist.ckpt
/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
torch.utils._pytree._register_pytree_node(
Seed set to 42
/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/huggingface_hub/file_download.py:1132: FutureWarning: resume_download is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use force_download=True.
warnings.warn(
Loading VIT
Loading VIT Done
Loading Q-Former
Loading Q-Former Done
Loading LLAMA
You are using the legacy behaviour of the <class 'transformers.models.llama.tokenization_llama.LlamaTokenizer'>. This means that tokens that come after special tokens will not be properly handled. We recommend you to read the related pull request available at huggingface/transformers#24565
Loading checkpoint shards: 100%|█| 2/2 [01:19<00:00, 39.9
Loading LLAMA Done
Load BLIP2-LLM Checkpoint: ../config/prerained_minigpt4_7b.pth
/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/torch/nn/modules/transformer.py:306: UserWarning: enable_nested_tensor is True, but self.use_nested_tensor is False because encoder_layer.norm_first was True
warnings.warn(f"enable_nested_tensor is True, but self.use_nested_tensor is False because {why_not_sparsity_fast_path}")
Loading pipeline components...: 0%| | 0/6 [00:00<?, ?it/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/diffusers/utils/outputs.py:63: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
torch.utils._pytree._register_pytree_node(
Loading pipeline components...: 100%|█| 6/6 [00:00<00:00,
Traceback (most recent call last):
File "/home/Intern1/Yun_SRP/MiniGPT-5/examples/playground.py", line 76, in
ax.imshow(image_out)
File "/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/matplotlib/init.py", line 1442, in inner
return func(ax, *map(sanitize_sequence, args), **kwargs)
File "/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/matplotlib/axes/_axes.py", line 5665, in imshow
im.set_data(X)
File "/home/Intern1/anaconda3/envs/minigpt5/lib/python3.9/site-packages/matplotlib/image.py", line 701, in set_data
raise TypeError("Image data of dtype {} cannot be converted to "
TypeError: Image data of dtype object cannot be converted to float