summaryrefslogtreecommitdiffstats
path: root/pipelines/stable_diffusion/no_check.py
blob: 6a6b33cdd42e41292f2d75cf4124a65f79970e92 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
from diffusers import ModelMixin, ConfigMixin
import torch


class NoCheck(ModelMixin, ConfigMixin):
    """Can be used in place of safety checker. Use responsibly and at your own risk."""

    def __init__(self):
        super().__init__()
        self.register_parameter(name='asdf', param=torch.nn.Parameter(torch.randn(3)))

    def forward(self, images=None, **kwargs):
        return images, [False]