Open sby-a-izumi opened 1 week ago
I believe I encountered a similar issue and it had something to do with the name of the files inside my weights folder, could you share a screenshot of your weights folder and the content inside its subfolders (icon_caption_blip2, icon_caption_florence,icon_detect)
Hi @AkimT13, are you running on cpu or gpu machine?
I switched everything to run on mps so it should be using my gpu. However in order to get it to run I had to change filename: pytorch_model-00001-of-00002.bin to be pytorch_model.bin inside the weights/icon_caption_blip2 folder. Theres still a file called pytorch_model-00002-of-00002.bin that is probably not getting used but it still runs fine anyway. Although it says this: "Some weights of Blip2ForConditionalGeneration were not initialized from the model checkpoint at weights/icon_caption_blip2 and are newly initialized because the shapes did not match: " and then a long list of torch's being initialized. Do you have any ideas on how to fix this? p.s I am very new to this kind of stuff :)
-Thanks, Akim
After setting up the environment by following the tutorial, I ran
gradio_demo.py
and encountered the following error: