update

app.py

@@ -216,17 +216,18 @@ def generate(
             seed=seed,
         ).images[0]
     elif isinstance(pipeline, ConceptrolIPAdapterPlus) or isinstance(pipeline, ConceptrolIPAdapterPlusXL):
-        images = pipeline.generate(
-            prompt=prompt,
-            pil_images=[image],
-            subjects=[subject],
-            num_samples=1,
-            num_inference_steps=50,
-            scale=condition_scale,
-            negative_prompt=negative_prompt,
-            control_guidance_start=control_guidance_start,
-            seed=seed,
-        )[0]
+        with torch.cuda.amp.autocast():
+            images = pipeline.generate(
+                prompt=prompt,
+                pil_images=[image],
+                subjects=[subject],
+                num_samples=1,
+                num_inference_steps=50,
+                scale=condition_scale,
+                negative_prompt=negative_prompt,
+                control_guidance_start=control_guidance_start,
+                seed=seed,
+            )[0]
     else:
         raise TypeError("Unsupported Pipeline")