Upload SDG pipeline + classifier weights
Browse files
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 |
|