thwin27
/

Aria-sequential_mlp-bnb_nf4

Image-Text-to-Text

text-generation

4-bit precision

Model card Files Files and versions Community

thwin27 commited on 15 days ago

Commit

163b4d2

•

1 Parent(s): 9aa49dd

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ Run this model with:
 import requests
 import torch
 from PIL import Image
-from transformers import AutoModelForCausalLM, AutoProcessor
 torch.cuda.set_device(0)
 model_id_or_path = "thwin27/Aria-sequential_mlp-bnb_nf4"

 import requests
 import torch
 from PIL import Image
+from transformers import AutoModelForCausalLM, AutoProcessor, BitsAndBytesConfig
 torch.cuda.set_device(0)
 model_id_or_path = "thwin27/Aria-sequential_mlp-bnb_nf4"