myduy commited on
Commit
711ddc7
·
verified ·
1 Parent(s): 2f06106

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ import torch
35
  from transformers import AutoTokenizer, AutoModel
36
 
37
  # 1. Configuration
38
- repo_id = "myduy/dlm-vi2en-checkpoint-85500"
39
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
40
 
41
  # 2. Load Model & Tokenizer
 
35
  from transformers import AutoTokenizer, AutoModel
36
 
37
  # 1. Configuration
38
+ repo_id = "myduy/dlm-vi2en"
39
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
40
 
41
  # 2. Load Model & Tokenizer