Hugging Face's logo

OSTswiss
/
LiteViT5

PyTorch
litevit5
custom_code
Model card Files
xet
Community
4bi commited on
Commit
614ea0a
·
1 Parent(s): 506372e

changed name

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -33,7 +33,7 @@ from transformers import AutoModel, AutoTokenizer
33
  from transformers import SiglipProcessor
34
 
35
  # Load the model
36
- model = AutoModel.from_pretrained("LiteVit5/model", trust_remote_code=True)
37
 
38
  # Load tokenizer and processor
39
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
@@ -50,7 +50,7 @@ from transformers import AutoModel, AutoTokenizer
50
  from transformers import SiglipProcessor
51
 
52
  # Load the model
53
- model = AutoModel.from_pretrained("LiteVit5/model", trust_remote_code=True, device_map="auto")
54
 
55
  # Load tokenizer and processor
56
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
 
33
  from transformers import SiglipProcessor
34
 
35
  # Load the model
36
+ model = AutoModel.from_pretrained("OSTswiss/LiteViT5", trust_remote_code=True)
37
 
38
  # Load tokenizer and processor
39
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
 
50
  from transformers import SiglipProcessor
51
 
52
  # Load the model
53
+ model = AutoModel.from_pretrained("OSTswiss/LiteViT5", trust_remote_code=True, device_map="auto")
54
 
55
  # Load tokenizer and processor
56
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")