Update README.md
Browse filesset use_fast=False
README.md
CHANGED
@@ -18,7 +18,7 @@ import torch
|
|
18 |
import requests
|
19 |
|
20 |
# Load model, tokenizer, image_processor
|
21 |
-
tokenizer = AutoTokenizer.from_pretrained("jxu124/TiO")
|
22 |
image_processor = AutoImageProcessor.from_pretrained("jxu124/TiO")
|
23 |
model = AutoModel.from_pretrained("jxu124/TiO", trust_remote_code=True)
|
24 |
model = model.to(torch.float16).cuda() # It will be faster when using float16.
|
|
|
18 |
import requests
|
19 |
|
20 |
# Load model, tokenizer, image_processor
|
21 |
+
tokenizer = AutoTokenizer.from_pretrained("jxu124/TiO", use_fast=False)
|
22 |
image_processor = AutoImageProcessor.from_pretrained("jxu124/TiO")
|
23 |
model = AutoModel.from_pretrained("jxu124/TiO", trust_remote_code=True)
|
24 |
model = model.to(torch.float16).cuda() # It will be faster when using float16.
|