taegyeonglee commited on
Commit
75353d5
·
verified ·
1 Parent(s): 1810b0b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -96,8 +96,8 @@ texts = [
96
  ]
97
 
98
 
99
- tokenizer = AutoTokenizer.from_pretrained(repo_or_dir, local_files_only=True)
100
- model = AutoModel.from_pretrained(repo_or_dir, trust_remote_code=True, local_files_only=True)
101
  device = "cuda" if torch.cuda.is_available() else "cpu"
102
  model.to(device).eval()
103
 
 
96
  ]
97
 
98
 
99
+ tokenizer = AutoTokenizer.from_pretrained(repo_or_dir)
100
+ model = AutoModel.from_pretrained(repo_or_dir, trust_remote_code=True)
101
  device = "cuda" if torch.cuda.is_available() else "cpu"
102
  model.to(device).eval()
103