basimazam commited on
Commit
59fe1aa
·
verified ·
1 Parent(s): ee0531a

Upload folder using huggingface_hub

Browse files
.gitignore CHANGED
@@ -23,3 +23,6 @@ Thumbs.db
23
  # Misc
24
  *.log
25
  outputs/
 
 
 
 
23
  # Misc
24
  *.log
25
  outputs/
26
+
27
+ !classifiers/
28
+ !classifiers/*.pth
classifiers/safety_classifier_1280.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3442ac01740c67c62ec1b5dabb59cdd70c0fe48f3245413a15d53338cdf90812
3
+ size 90034677
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("classifiers/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("safety_classifier_1280.pth")
81
 
82
  class SDGOutput(BaseOutput):
83
  images: List