kekekeke commited on
Commit
d2b226f
·
verified ·
1 Parent(s): 2b33fa9

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. .gitattributes +0 -1
  2. README.md +5 -5
.gitattributes CHANGED
@@ -1,2 +1 @@
1
  *.safetensors filter=lfs diff=lfs merge=lfs -text
2
- tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
1
  *.safetensors filter=lfs diff=lfs merge=lfs -text
 
README.md CHANGED
@@ -30,10 +30,10 @@ import torch
30
  from qwen_vl_utils import process_vision_info
31
  from transformers import AutoModel, AutoProcessor
32
 
33
- model_path = "/path/to/WeMM-Embedding-9B"
34
- processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
35
  model = AutoModel.from_pretrained(
36
- model_path, trust_remote_code=True, dtype=torch.bfloat16
37
  ).cuda().eval()
38
 
39
  messages = [{"role": "user", "content": [
@@ -75,8 +75,8 @@ Use any subset of the content items to encode text, image, or video independentl
75
  ```python
76
  from wemm_sentence_transformers import load_wemm_sentence_transformer
77
 
78
- model_path = "/path/to/WeMM-Embedding-9B"
79
- model = load_wemm_sentence_transformer(model_path, device="cuda:0")
80
  inputs = [
81
  "A dog is running on a beach.",
82
  {"image": "/path/to/image.jpg", "text": "Represent this image."},
 
30
  from qwen_vl_utils import process_vision_info
31
  from transformers import AutoModel, AutoProcessor
32
 
33
+ model_id = "tencent/WeMM-Embedding-9B"
34
+ processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True)
35
  model = AutoModel.from_pretrained(
36
+ model_id, trust_remote_code=True, dtype=torch.bfloat16
37
  ).cuda().eval()
38
 
39
  messages = [{"role": "user", "content": [
 
75
  ```python
76
  from wemm_sentence_transformers import load_wemm_sentence_transformer
77
 
78
+ model_id = "tencent/WeMM-Embedding-9B"
79
+ model = load_wemm_sentence_transformer(model_id, device="cuda:0")
80
  inputs = [
81
  "A dog is running on a beach.",
82
  {"image": "/path/to/image.jpg", "text": "Represent this image."},