Tagging Of Images Taking Longer Than Usual (Dataset Maker)
LmaoPro23 opened this issue · 3 comments
LmaoPro23 commented
Its taking longer time to tag the images (since today), it used to tag 400 images in a few minutes and now it takes around 30 mins...
hollowstrawberry commented
It might be running on CPU after the latest colab update broke cuda. Unfortunately I don't know how to approach this problem. Waiting for further comments.
hollowstrawberry commented
Were you able to solve the issue?
LmaoPro23 commented
Were you able to solve the issue?
changing this seems to have fixed it.. and now its even faster than before :3
print("\n🏭 Installing dependencies...\n")
!pip -q install --upgrade tensorflow huggingface-hub==0.12.0 accelerate==0.15.0 transformers==4.26.0 diffusers[torch]==0.10.2 einops==0.6.0 safetensors==0.2.6 torchvision albumentations