debugging app.py

app.py

@@ -154,7 +154,7 @@ def create_app():
             filename="FLUX/it-blender.bin",
             token=token
         )
-        print("pipe._execution_device",pipe._execution_device)
+        
         pretrained_blended_attn_weights = torch.load(model_path, map_location=pipe._execution_device)
 
         key_changed_blended_attn_weights = {}
@@ -162,7 +162,7 @@ def create_app():
             block_idx = int(key.split(".")[0]) - 21
             k_or_v = key.split("_")[2]
             changed_key = f'single_transformer_blocks.{block_idx}.attn.processor.blended_attention_{k_or_v}_proj.weight'
-            key_changed_blended_attn_weights[changed_key] = value.to(dtype)
+            key_changed_blended_attn_weights[changed_key] = value.to(dtype=dtype, device="cuda")
             
         missing_keys, unexpected_keys = pipe.transformer.load_state_dict(key_changed_blended_attn_weights, strict=False)
 
@@ -184,5 +184,4 @@ def create_app():
 
 if __name__ == "__main__":
     demo = create_app()
-    # demo.launch(debug=True, queue=False)
     demo.launch(debug=True, ssr_mode=False)