Spaces:

josuelmet
/

Boomerang_Stable_Diffusion

Runtime error

josuelmet commited on Oct 13, 2022

Commit

a3f6bed

•

1 Parent(s): f26d531

Update boomerang.py

Files changed (1) hide show

boomerang.py CHANGED Viewed

@@ -251,6 +251,7 @@ def call(
     return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
-pipe.unet.forward = MethodType(forward, pipe.unet)
-pipe.safety_checker.forward = MethodType(safety_forward, pipe.safety_checker)
 type(pipe).__call__ = call

     return StableDiffusionPipelineOutput(images=image, nsfw_content_detected=has_nsfw_concept)
+if COLAB:
+    pipe.unet.forward = MethodType(forward, pipe.unet)
+    pipe.safety_checker.forward = MethodType(safety_forward, pipe.safety_checker)
 type(pipe).__call__ = call