standardmodelbio
/

SMB-RAD-Encoder-v1

smb_vision_model

Model card Files Files and versions

zekaic commited on 4 days ago

Commit

8610c69

·

verified ·

1 Parent(s): cd83b6d

Update README.md

Files changed (1) hide show

README.md +2 -4

README.md CHANGED Viewed

@@ -104,13 +104,11 @@ model = AutoModel.from_pretrained(
     trust_remote_code=True,
     dtype=torch.bfloat16,
     attn_implementation="flash_attention_2",
-)
 model.to("cuda")
 # Encode features
-encoded_patches, deepstack_features = model.forward_features(
-    images.to("cuda"), grid_thw=grid_thw.to("cuda")
-)
 print(encoded_patches.shape)
 # (800, 1152)
 ```

     trust_remote_code=True,
     dtype=torch.bfloat16,
     attn_implementation="flash_attention_2",
+).encoder
 model.to("cuda")
 # Encode features
+encoded_patches, deepstack_features = model(images.to("cuda"), grid_thw=grid_thw.to("cuda"))
 print(encoded_patches.shape)
 # (800, 1152)
 ```