Upload folder using huggingface_hub
Browse files
safe_diffusion_guidance.py
CHANGED
|
@@ -77,7 +77,7 @@ def load_classifier_1280(
|
|
| 77 |
|
| 78 |
def pick_weights_for_pipe(pipe) -> str:
|
| 79 |
# Always use the standard path inside the repo
|
| 80 |
-
return _resolve_repo_path("safety_classifier_1280.pth")
|
| 81 |
|
| 82 |
class SDGOutput(BaseOutput):
|
| 83 |
images: List
|
|
|
|
| 77 |
|
| 78 |
def pick_weights_for_pipe(pipe) -> str:
|
| 79 |
# Always use the standard path inside the repo
|
| 80 |
+
return _resolve_repo_path("classifiers/safety_classifier_1280.pth")
|
| 81 |
|
| 82 |
class SDGOutput(BaseOutput):
|
| 83 |
images: List
|