jxu124
/

TiO

Feature Extraction

Model card Files Files and versions

jxu124 commited on Sep 25, 2023

Commit

cfeea93

·

1 Parent(s): 34373e1

Update README.md

set use_fast=False

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -18,7 +18,7 @@ import torch
 import requests
 # Load model, tokenizer, image_processor
-tokenizer = AutoTokenizer.from_pretrained("jxu124/TiO")
 image_processor = AutoImageProcessor.from_pretrained("jxu124/TiO")
 model = AutoModel.from_pretrained("jxu124/TiO", trust_remote_code=True)
 model = model.to(torch.float16).cuda()  # It will be faster when using float16.

 import requests
 # Load model, tokenizer, image_processor
+tokenizer = AutoTokenizer.from_pretrained("jxu124/TiO", use_fast=False)
 image_processor = AutoImageProcessor.from_pretrained("jxu124/TiO")
 model = AutoModel.from_pretrained("jxu124/TiO", trust_remote_code=True)
 model = model.to(torch.float16).cuda()  # It will be faster when using float16.