Amitz244 commited on
Commit
fb43af1
·
verified ·
1 Parent(s): 196e498

Update modeling.py

Browse files
Files changed (1) hide show
  1. modeling.py +4 -4
modeling.py CHANGED
@@ -46,8 +46,8 @@ class clip_lora_model(nn.Module):
46
  self.model = get_peft_model(encoder, config)
47
 
48
  def forward(self, x,feat=None,flags=None):
49
- sub_voxels = self.model(x)
50
- image_embeds = sub_voxels[1]
51
- sub_voxels = self.proj(image_embeds)
52
- outputs = self.mlp(sub_voxels)
53
  return outputs
 
46
  self.model = get_peft_model(encoder, config)
47
 
48
  def forward(self, x,feat=None,flags=None):
49
+ model_outputs = self.model(x)
50
+ image_embeds = model_outputs[1]
51
+ model_outputs = self.proj(image_embeds)
52
+ outputs = self.mlp(model_outputs)
53
  return outputs