ZeroGPU

app.py

@@ -82,7 +82,15 @@ qwen2vl_processor = AutoProcessor.from_pretrained(
 )
 
 # 加载 connector 和 embedder 到 CPU
-connector = nn.Linear(3584, 4096).to(dtype).cpu()
+class Qwen2Connector(nn.Module):
+    def __init__(self, input_dim=3584, output_dim=4096):
+        super().__init__()
+        self.linear = nn.Linear(input_dim, output_dim)
+    
+    def forward(self, x):
+        return self.linear(x)
+
+connector = Qwen2Connector().to(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.replace('module.', ''): v.to(dtype) for k, v in connector_state.items()}