lmms-lab-encoder
/

onevision-encoder-large-lang

Image Feature Extraction

onevision_encoder

feature-extraction

Model card Files Files and versions

xiangan commited on Feb 10

Commit

f9f51fe

·

verified ·

1 Parent(s): bd417f2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,13 +36,13 @@ import torch
 # Load model and preprocessor
 model = AutoModel.from_pretrained(
-    "lmms-lab-encoder/onevision-encoder-large",
     trust_remote_code=True,
     attn_implementation="flash_attention_2"
 ).to("cuda").eval()
 preprocessor = AutoImageProcessor.from_pretrained(
-    "lmms-lab-encoder/onevision-encoder-large",
     trust_remote_code=True
 )

 # Load model and preprocessor
 model = AutoModel.from_pretrained(
+    "lmms-lab-encoder/onevision-encoder-large-lang",
     trust_remote_code=True,
     attn_implementation="flash_attention_2"
 ).to("cuda").eval()
 preprocessor = AutoImageProcessor.from_pretrained(
+    "lmms-lab-encoder/onevision-encoder-large-lang",
     trust_remote_code=True
 )