benchang1110
/

TaiVisionLM-base-v1

Image-Text-to-Text

text-generation

Model card Files Files and versions

benchang1110 commited on Aug 24, 2024

Commit

88773ae

·

verified ·

1 Parent(s): 6eef94f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -91,7 +91,7 @@ from PIL import Image
 import requests
 import torch
-config = AutoConfig.from_pretrained("benchang1110/TaiVisioLMn-base",trust_remote_code=True)
 processor = AutoProcessor.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True,torch_dtype=torch.float16,attn_implementation="sdpa").to('cuda')
 model.eval()

 import requests
 import torch
+config = AutoConfig.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True)
 processor = AutoProcessor.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True,torch_dtype=torch.float16,attn_implementation="sdpa").to('cuda')
 model.eval()