samwell commited on
Commit
6240e74
·
verified ·
1 Parent(s): ed4775c

Add fake cosmos_guardrail module

Browse files
Files changed (1) hide show
  1. cosmos_guardrail.py +14 -0
cosmos_guardrail.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Fake cosmos_guardrail module to satisfy diffusers import"""
2
+
3
+ class CosmosSafetyChecker:
4
+ def __init__(self):
5
+ pass
6
+
7
+ def __call__(self, frames, **kwargs):
8
+ return frames
9
+
10
+ def check_text_safety(self, text):
11
+ return True
12
+
13
+ def check_video_safety(self, frames):
14
+ return frames