Spaces:

shichen1231
/

illumination_composition_controlnet

Runtime error

shichen1231 commited on Jun 7, 2023

Commit

c74fa4c

1 Parent(s): 9f9752e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,8 +31,8 @@ def infer(
         ill=0.6,
 ):
-    conditioning_image = Image.fromarray(conditioning_image)
-    # conditioning_image = conditioning_image_raw.convert('L')
     g_cpu = torch.Generator()
     if seed == -1:
@@ -53,7 +53,8 @@ def infer(
     ).images[0]
-    del conditioning_image, conditioning_image_raw,seg_image
     gc.collect()
     return output_image

         ill=0.6,
 ):
+    conditioning_image_raw = Image.fromarray(conditioning_image)
+    conditioning_image = conditioning_image_raw.convert('L')
     g_cpu = torch.Generator()
     if seed == -1:
     ).images[0]
+    del conditioning_image, conditioning_image_raw
     gc.collect()
     return output_image