PEFT
Safetensors
vidore-experimental
vidore
manu commited on
Commit
9f79f9a
·
verified ·
1 Parent(s): a55b405

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- base_model: manu/ColSmolVLM-base
3
  library_name: peft
4
  tags:
5
  - vidore-experimental
@@ -56,12 +56,12 @@ from PIL import Image
56
  from colpali_engine.models import ColIdefics3, ColIdefics3Processor
57
 
58
  model = ColIdefics3.from_pretrained(
59
- "vidore/colqwen2-v1.0",
60
  torch_dtype=torch.bfloat16,
61
  device_map="cuda:0",
62
  attn_implementation="flash_attention_2" # or eager
63
  ).eval()
64
- processor = ColIdefics3Processor.from_pretrained("manu/colsmolvlm-alpha")
65
 
66
  # Your inputs
67
  images = [
 
1
  ---
2
+ base_model: vidore/ColSmolVLM-base
3
  library_name: peft
4
  tags:
5
  - vidore-experimental
 
56
  from colpali_engine.models import ColIdefics3, ColIdefics3Processor
57
 
58
  model = ColIdefics3.from_pretrained(
59
+ "vidore/colsmolvlm-alpha",
60
  torch_dtype=torch.bfloat16,
61
  device_map="cuda:0",
62
  attn_implementation="flash_attention_2" # or eager
63
  ).eval()
64
+ processor = ColIdefics3Processor.from_pretrained("vidore/colsmolvlm-alpha")
65
 
66
  # Your inputs
67
  images = [