main

gradio_app_asy.py

@@ -117,7 +117,7 @@ def infer(prompt, frame, seed=0):
 
     logger.info(f"Using seed: {seed}")
 
-    # ae.to("cpu")
+    ae.to("cpu")
     clip_l.to(device)
     t5xxl.to(device)
 
@@ -151,8 +151,8 @@ def infer(prompt, frame, seed=0):
     args = lambda: None
     args.frame_num = frame_num
 
-    # clip_l.to("cpu")
-    # t5xxl.to("cpu")
+    clip_l.to("cpu")
+    t5xxl.to("cpu")
 
     model.to(device)
 
@@ -182,12 +182,12 @@ def infer(prompt, frame, seed=0):
     # Decode the final image
     x = x.float()
     x = flux_utils.unpack_latents(x, packed_latent_height, packed_latent_width)
-    # model.to("cpu")
+    model.to("cpu")
     ae.to(device)
     with accelerator.autocast(), torch.no_grad():
         x = ae.decode(x)
     logger.debug("Latents decoded into image.")
-    # ae.to("cpu")
+    ae.to("cpu")
 
     # Convert the tensor to an image
     x = x.clamp(-1, 1)
@@ -270,6 +270,12 @@ with gr.Blocks() as demo:
             "sks6, 3*3 puzzle of 9 sub-images, step-by-step portrait painting process, woman with blonde curly hair",
             1062070717
         ],
+        [
+            "9 frame",
+            "Sand art",
+            "sks8, 3*3 puzzle of 9 sub-images, step-by-step description of sand art creation, <image-1>: The outline of a classic pirate ship is drawn, capturing its sails and hull. <image-2>: Basic shapes of the ship’s structure and masts are added, defining its adventurous form. <image-3>: Details of the sails and rigging begin to appear, adding complexity. <image-4>: Shadows and highlights enhance the ship’s three-dimensional appearance. <image-5>: The ship’s deck and cannons are refined, giving it character. <image-6>: Additional elements like waves and seagulls are added for movement. <image-7>: A backdrop of a stormy sea with dark clouds is introduced, adding drama. <image-8>: Further details like lightning and crashing waves are sketched for intensity. <image-9>: Final touches include vibrant blues and grays, completing the thrilling pirate ship scene.",
+            641262478
+        ],
     ]
 
     gr.Examples(