EPFL-VILAB
/

VPE-ViPer

Image-Text-to-Text

Model card Files Files and versions

Sogand Salehi commited on Jul 22, 2024

Commit

ea78f78

·

verified ·

1 Parent(s): bf5e04b

Update README.md

Files changed (1) hide show

README.md +3 -0

README.md CHANGED Viewed

@@ -24,6 +24,9 @@ For install instructions, please see https://github.com/EPFL-VILAB/ViPer.
 This model can be loaded from Hugging Face Hub as follows:
 ```python
 model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b").to(device)
 model = PeftModel.from_pretrained(model, "EPFL-VILAB/VPE-ViPer").to(device)
 ```

 This model can be loaded from Hugging Face Hub as follows:
 ```python
+from transformers import AutoModelForVision2Seq
+from peft import PeftModel
 model = AutoModelForVision2Seq.from_pretrained("HuggingFaceM4/idefics2-8b").to(device)
 model = PeftModel.from_pretrained(model, "EPFL-VILAB/VPE-ViPer").to(device)
 ```