cosmos-predict2-endpoint / cosmos_guardrail.py
samwell's picture
Add fake cosmos_guardrail module
6240e74 verified
"""Fake cosmos_guardrail module to satisfy diffusers import"""
class CosmosSafetyChecker:
def __init__(self):
pass
def __call__(self, frames, **kwargs):
return frames
def check_text_safety(self, text):
return True
def check_video_safety(self, frames):
return frames