debugging app.py

app.py

@@ -142,12 +142,11 @@ def create_app():
         blended_attn_procs = {}
         for name, _ in pipe.transformer.attn_processors.items():
             if "single" in name:
-                blended_attn_procs[name] = FluxBlendedAttnProcessor2_0(3072, ba_scale=scale, num_ref=1)
+                blended_attn_procs[name] = FluxBlendedAttnProcessor2_0(3072, ba_scale=scale, num_ref=1).to(device="cuda", dtype=dtype)
             else:
                 blended_attn_procs[name] = pipe.transformer.attn_processors[name]
 
         pipe.transformer.set_attn_processor(blended_attn_procs)
-        pipe.to(dtype)
 
         model_path = hf_hub_download(
             repo_id="WonwoongCho/IT-Blender",