Image-Text-to-Text
PEFT
Safetensors
Sogand Salehi commited on
Commit
be14985
·
verified ·
1 Parent(s): 8fddf3d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -27,8 +27,8 @@ This model can be loaded from Hugging Face Hub as follows:
27
  from transformers import AutoProcessor, BitsAndBytesConfig, AutoModelForVision2Seq
28
  from peft import PeftModel
29
 
30
- model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b").to(device)
31
- model = PeftModel.from_pretrained(model, "EPFL-VILAB/Metric-ViPer").to(device)
32
  ```
33
 
34
  Please see https://github.com/EPFL-VILAB/ViPer for more detailed instructions.
 
27
  from transformers import AutoProcessor, BitsAndBytesConfig, AutoModelForVision2Seq
28
  from peft import PeftModel
29
 
30
+ model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b")
31
+ model = PeftModel.from_pretrained(model, "EPFL-VILAB/Metric-ViPer")
32
  ```
33
 
34
  Please see https://github.com/EPFL-VILAB/ViPer for more detailed instructions.