Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,6 @@ image2 = "image2.jpg"
|
|
| 58 |
images = [Image.open(image1), Image.open(image2)]
|
| 59 |
|
| 60 |
# load processor and model
|
| 61 |
-
# from mantis.models.mllava import MLlavaProcessor, LlavaForConditionalGeneration
|
| 62 |
from mantis.models.mfuyu import MFuyuForCausalLM, MFuyuProcessor
|
| 63 |
processor = MFuyuProcessor.from_pretrained("TIGER-Lab/Mantis-8B-Fuyu")
|
| 64 |
attn_implementation = None # or "flash_attention_2"
|
|
|
|
| 58 |
images = [Image.open(image1), Image.open(image2)]
|
| 59 |
|
| 60 |
# load processor and model
|
|
|
|
| 61 |
from mantis.models.mfuyu import MFuyuForCausalLM, MFuyuProcessor
|
| 62 |
processor = MFuyuProcessor.from_pretrained("TIGER-Lab/Mantis-8B-Fuyu")
|
| 63 |
attn_implementation = None # or "flash_attention_2"
|