English
qtnx commited on
Commit
b534628
1 Parent(s): 029d226
Files changed (1) hide show
  1. __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("./checkpoints/llama-3/checkpoint-2400/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()
 
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()