Initial Commit

app.py

@@ -114,7 +114,7 @@ class FluxInterface:
         connector = Qwen2Connector().to(self.dtype).cpu()
         connector_path = os.path.join(MODEL_CACHE_DIR, "qwen2-vl/connector.pt")
         connector_state = torch.load(connector_path, map_location='cpu')
-        connector_state = {k: v.to(self.dtype) for k, v in connector_state.items()}
+        connector_state = {k.replace('module.', ''): v.to(self.dtype) for k, v in connector_state.items()}
         connector.load_state_dict(connector_state)
         
         self.t5_context_embedder = nn.Linear(4096, 3072).to(self.dtype).cpu()