mirror of
https://github.com/sd-webui/stable-diffusion-webui.git
synced 2024-12-15 07:12:58 +03:00
12 lines
378 B
Python
12 lines
378 B
Python
|
from diffusers import ModelMixin
|
||
|
|
import torch
|
||
|
|
|
||
|
|
class NoCheck(ModelMixin):
|
||
|
|
"""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]
|