Upload folder using huggingface_hub

safe_diffusion_guidance.py

@@ -3,8 +3,17 @@ import torch
 from typing import Optional, List
 from diffusers import DiffusionPipeline, StableDiffusionPipeline
 from diffusers.utils import BaseOutput
+# at the top of safe_diffusion_guidance.py
 
-from utils.adaptive_classifiers import load_classifier_1280, pick_weights_for_pipe
+import os, sys
+FILE_DIR = os.path.dirname(__file__)
+UTILS_DIR = os.path.join(FILE_DIR, "utils")
+if UTILS_DIR not in sys.path:
+    sys.path.insert(0, UTILS_DIR)
+
+from adaptive_classifiers import load_classifier_1280, pick_weights_for_pipe
+
+# from utils.adaptive_classifiers import load_classifier_1280, pick_weights_for_pipe
 
 class SDGOutput(BaseOutput):
     images: List