• Suraj Patil's avatar
    Add safety module (#213) · 65ea7d6b
    Suraj Patil authored
    * add SafetyChecker
    
    * better name, fix checker
    
    * add checker in main init
    
    * remove from main init
    
    * update logic to detect pipeline module
    
    * style
    
    * handle all safety logic in safety checker
    
    * draw text
    
    * can't draw
    
    * small fixes
    
    * treat special care as nsfw
    
    * remove commented lines
    
    * update safety checker
    65ea7d6b
safety_checker.py 3.28 KB