chunmeifeng / DiffTPT

【ICCV 2023】Diverse Data Augmentation with Diffusions for Effective Test-time Prompt Tuning
56 stars 3 forks source link

'bool' object is not iterable #6

Open AlyssaYoung opened 8 months ago

AlyssaYoung commented 8 months ago

When I ran the command for Stable Diffusion based data augmentation, I got this error 'bool' object is not iterable. Then I annotated the pipe.safety_checker = lambda images, clip_input: (images, False) in image_generator_R.py. The error was not reported, but sometimes I got an all-black-generated image.

Do you have some advice for the above problem?

chunmeifeng commented 8 months ago

Hi, you can refer https://github.com/CompVis/stable-diffusion/issues/239 for more detail, maybe you can try a different GPU or SD version.

Pinci Yang @.***> 于2024年2月28日周三 16:02写道:

When I ran the command for Stable Diffusion based data augmentation, I got this error 'bool' object is not iterable. Then I annotated the pipe.safety_checker = lambda images, clip_input: (images, False) in image_generator_R.py. The error was not reported, but sometimes I got an all-black-generated image.

Do you have some advice for the above problem?

— Reply to this email directly, view it on GitHub https://github.com/chunmeifeng/DiffTPT/issues/6, or unsubscribe https://github.com/notifications/unsubscribe-auth/AR75XNYW3V3XV72QICQDQUDYV3QBZAVCNFSM6AAAAABD5QTUXCVHI2DSMVQWIX3LMV43ASLTON2WKOZSGE2TQMZUGU2TENQ . You are receiving this because you are subscribed to this thread.Message ID: @.***>

AlyssaYoung commented 8 months ago

Thanks a lot for your reply. I guess it maybe due to the 'nsfw' filtering.