stanrom
/

internlm-xcomposer2-7b

Text Generation

internlmxcomposer2

feature-extraction

Model card Files Files and versions Community

stanrom commited on May 30

Commit

6b88327

•

1 Parent(s): c46a3ad

Update build_mlp.py

Files changed (1) hide show

build_mlp.py +2 -1

build_mlp.py CHANGED Viewed

@@ -130,7 +130,8 @@ class CLIPVisionTower(nn.Module):
  image_features.append(image_feature)
  else:
  image_forward_outs = self.vision_tower(
- images.to(device=self.device, dtype=self.dtype),
  output_hidden_states=True)
  image_features = self.feature_select(image_forward_outs).to(
  images.dtype)

  image_features.append(image_feature)
  else:
  image_forward_outs = self.vision_tower(
+ # images.to(device=self.device, dtype=self.dtype),
+ images,
  output_hidden_states=True)
  image_features = self.feature_select(image_forward_outs).to(
  images.dtype)