cosmos_guardrail.py

"""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