Hugging Face's logo

benchang1110
/
TaiVisionLM-base-v1

Image-Text-to-Text
Transformers
Safetensors
Chinese
taivisionlm
text-generation
conversational
custom_code
Model card Files
xet
Community
1
benchang1110 commited on
Commit
99a353f
·
verified ·
1 Parent(s): 94a969c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -91,9 +91,9 @@ from PIL import Image
91
  import requests
92
  import torch
93
 
94
- config = AutoConfig.from_pretrained("benchang1110/TaiVision-base",trust_remote_code=True)
95
- processor = AutoProcessor.from_pretrained("benchang1110/TaiVision-base",trust_remote_code=True)
96
- model = AutoModelForCausalLM.from_pretrained("benchang1110/TaiVision-base",trust_remote_code=True,torch_dtype=torch.float16,attn_implementation="sdpa").to('cuda')
97
  model.eval()
98
  url = "https://media.wired.com/photos/598e35fb99d76447c4eb1f28/master/pass/phonepicutres-TA.jpg"
99
  image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
@@ -114,9 +114,9 @@ from PIL import Image
114
  import requests
115
  import torch
116
 
117
- config = AutoConfig.from_pretrained("benchang1110/TaiVision-base",trust_remote_code=True)
118
- processor = AutoProcessor.from_pretrained("benchang1110/TaiVision-base",trust_remote_code=True)
119
- model = AutoModelForCausalLM.from_pretrained("benchang1110/TaiVision-base",trust_remote_code=True,torch_dtype=torch.float16,attn_implementation="sdpa").to('cuda')
120
  model.eval()
121
  url = "https://media.wired.com/photos/598e35fb99d76447c4eb1f28/master/pass/phonepicutres-TA.jpg"
122
  image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 
91
  import requests
92
  import torch
93
 
94
+ config = AutoConfig.from_pretrained("benchang1110/TaiVisioLMn-base",trust_remote_code=True)
95
+ processor = AutoProcessor.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True)
96
+ model = AutoModelForCausalLM.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True,torch_dtype=torch.float16,attn_implementation="sdpa").to('cuda')
97
  model.eval()
98
  url = "https://media.wired.com/photos/598e35fb99d76447c4eb1f28/master/pass/phonepicutres-TA.jpg"
99
  image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 
114
  import requests
115
  import torch
116
 
117
+ config = AutoConfig.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True)
118
+ processor = AutoProcessor.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True)
119
+ model = AutoModelForCausalLM.from_pretrained("benchang1110/TaiVisionLM-base",trust_remote_code=True,torch_dtype=torch.float16,attn_implementation="sdpa").to('cuda')
120
  model.eval()
121
  url = "https://media.wired.com/photos/598e35fb99d76447c4eb1f28/master/pass/phonepicutres-TA.jpg"
122
  image = Image.open(requests.get(url, stream=True).raw).convert("RGB")