basimazam commited on
Commit
c44a3aa
·
verified ·
1 Parent(s): 1dec584

Upload SDG pipeline + classifier weights

Browse files
Files changed (1) hide show
  1. safe_diffusion_guidance.py +1 -1
safe_diffusion_guidance.py CHANGED
@@ -124,7 +124,7 @@ class SafeDiffusionGuidance(DiffusionPipeline):
124
  and applies mid-UNet classifier-guided denoising for safety.
125
  """
126
 
127
- def __init__(self): # IMPORTANT: no **kwargs (diffusers inspects this)
128
  super().__init__()
129
  self.base_pipe_ = None # lazy cache
130
 
 
124
  and applies mid-UNet classifier-guided denoising for safety.
125
  """
126
 
127
+ def __init__(self,**kwargs): # IMPORTANT: no **kwargs (diffusers inspects this)
128
  super().__init__()
129
  self.base_pipe_ = None # lazy cache
130