qihoo360
/

Inner-Adaptor-Architecture

Image-Text-to-Text

text-generation

Model card Files Files and versions

qingshan777 commited on Sep 3, 2024

Commit

ed8ecf2

·

verified ·

1 Parent(s): 3b5895f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 from PIL import Image
-checkpoint = "qihoo360/iaa-14-hf"
 model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='cuda', trust_remote_code=True).eval()
 tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)

 import torch
 from PIL import Image
+checkpoint = "qihoo360/Inner-Adaptor-Architecture"
 model = AutoModelForCausalLM.from_pretrained(checkpoint, torch_dtype=torch.float16, device_map='cuda', trust_remote_code=True).eval()
 tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True)