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()
|