Shuang59 commited on
Commit
27c35a5
β€’
1 Parent(s): 392b502

Update safety_checker.py

Browse files
Files changed (1) hide show
  1. safety_checker.py +3 -0
safety_checker.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import numpy as np
2
  import torch
3
  import torch.nn as nn
 
1
+ """
2
+ file copy from diffusion library from Huggingface: https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/safety_checker.py
3
+ """
4
  import numpy as np
5
  import torch
6
  import torch.nn as nn