Transformers
Safetensors
English
German
qwen2_vl
multimodal-embedding
text-generation-inference
Inference Endpoints
tattrongvu commited on
Commit
9d7f124
·
verified ·
1 Parent(s): d1f9b3f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -12,7 +12,7 @@ base_model:
12
  tags:
13
  - vidore
14
  - multimodal-embedding
15
- library_name: peft
16
  ---
17
  # ColQwen2-7B: Visual Retriever based on Qwen2-VL-7B-Instruct with ColBERT strategy
18
 
@@ -65,11 +65,11 @@ from PIL import Image
65
  from colpali_engine.models import ColQwen2, ColQwen2Processor
66
 
67
  model = ColQwen2.from_pretrained(
68
- "tsystems/colqwen2-7b-v1.0",
69
  torch_dtype=torch.bfloat16,
70
  device_map="cuda:0", # or "mps" if on Apple Silicon
71
  ).eval()
72
- processor = ColQwen2Processor.from_pretrained("tsystems/colqwen2-7b-v1.0")
73
 
74
  # Your inputs
75
  images = [
 
12
  tags:
13
  - vidore
14
  - multimodal-embedding
15
+ library_name: transformers
16
  ---
17
  # ColQwen2-7B: Visual Retriever based on Qwen2-VL-7B-Instruct with ColBERT strategy
18
 
 
65
  from colpali_engine.models import ColQwen2, ColQwen2Processor
66
 
67
  model = ColQwen2.from_pretrained(
68
+ "tsystems/colqwen2-7b-v1.0-merged",
69
  torch_dtype=torch.bfloat16,
70
  device_map="cuda:0", # or "mps" if on Apple Silicon
71
  ).eval()
72
+ processor = ColQwen2Processor.from_pretrained("tsystems/colqwen2-7b-v1.0-merged")
73
 
74
  # Your inputs
75
  images = [