Spaces:

ktrndy
/

diffusion-image-gen

Running

ktrndy commited on Feb 15

Commit

7cdef21

verified ·

1 Parent(s): 5280b61

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,6 @@ def infer(
                 torch_dtype=torch_dtype
             )
         else:
-            controlnet_mode == "edge_detection":
             controlnet = ControlNetModel.from_pretrained(
                 "lllyasviel/sd-controlnet-canny",
                 cache_dir="./models_cache",

                 torch_dtype=torch_dtype
             )
         else:
             controlnet = ControlNetModel.from_pretrained(
                 "lllyasviel/sd-controlnet-canny",
                 cache_dir="./models_cache",