fix path
Browse files- __main__.py +1 -1
__main__.py
CHANGED
@@ -100,7 +100,7 @@ class ProjectionModule(nn.Module):
|
|
100 |
|
101 |
def load_projection_module(mm_hidden_size=1152, hidden_size=4096, device='cuda'):
|
102 |
projection_module = ProjectionModule(mm_hidden_size, hidden_size)
|
103 |
-
checkpoint = torch.load("./
|
104 |
checkpoint = {k.replace("mm_projector.", ""): v for k, v in checkpoint.items()}
|
105 |
projection_module.load_state_dict(checkpoint)
|
106 |
projection_module = projection_module.to(device).half()
|
|
|
100 |
|
101 |
def load_projection_module(mm_hidden_size=1152, hidden_size=4096, device='cuda'):
|
102 |
projection_module = ProjectionModule(mm_hidden_size, hidden_size)
|
103 |
+
checkpoint = torch.load("./mm_projector.bin")
|
104 |
checkpoint = {k.replace("mm_projector.", ""): v for k, v in checkpoint.items()}
|
105 |
projection_module.load_state_dict(checkpoint)
|
106 |
projection_module = projection_module.to(device).half()
|