dmarx / video-killed-the-radio-star

Notebook and tools for end-to-end automation of music video production with generative AI
https://colab.research.google.com/github/dmarx/video-killed-the-radio-star/blob/main/Video_Killed_The_Radio_Star_Defusion.ipynb#scrollTo=oPbeyWtesAoh
MIT License
198 stars 35 forks source link

Disable content filtering/option to disable censorship #26

Closed ChrisChowen closed 2 years ago

ChrisChowen commented 2 years ago

Hi, I noticed that since it's running from your library there is no easy way to add the necessary code to disable the content filters. This was the main reason I wanted to get away from using the API.

I'm used to adding pipe.safety_checker = lambda images, **kwargs: (images, False) to most models to get around the issues but I can't find where to add this in here.

It seemingly finds a way to block a few images out of every song generated which is a fairly annoying issue. At the very least, it would be good that when it detects an "innapropriate" image that it keeps using new seeds until it gets a suitable one.

ChrisChowen commented 2 years ago

I noticed you may have run out of time last night trying to get that second option working so it looks like you're already on this, I tried to fix but my 6 months of Python experience isn't enough to be any use! Also removing the filter would be preferred over regenerating since it's wastes resources.

dmarx commented 2 years ago

calling this wontfix for now, will probably revisit in near future