Spaces:

KarthikAI
/

Sticker_Diffusion

Running

KarthikAI commited on 1 day ago

Commit

779803e

verified ·

1 Parent(s): f2b2714

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -21,7 +21,7 @@ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     MODEL_ID,
     torch_dtype=torch.float32,
     cache_dir=MODEL_CACHE,
-    safety_checker=None,  # Disable for demo/testing; enable in prod
 ).to(DEVICE)
 def generate_sticker(input_image, prompt):

     MODEL_ID,
     torch_dtype=torch.float32,
     cache_dir=MODEL_CACHE,
+    # safety_checker=None,  # Disable for demo/testing; enable in prod
 ).to(DEVICE)
 def generate_sticker(input_image, prompt):