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

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. safe_diffusion_guidance.py +1 -1
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