myduy commited on
Commit
b324875
·
verified ·
1 Parent(s): 16a1726

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -58,7 +58,7 @@ Nguồn dữ liệu được sử dụng là từ:
58
  import torch
59
  from transformers import AutoModel, AutoTokenizer
60
 
61
- model_path = "myduy/dream-diffusion-ner-recovery"
62
  model = AutoModel.from_pretrained(model_path, torch_dtype=torch.bfloat16, trust_remote_code=True)
63
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
64
  model = model.to("cuda").eval()
 
58
  import torch
59
  from transformers import AutoModel, AutoTokenizer
60
 
61
+ model_path = "myduy/diffusion-medical-ner"
62
  model = AutoModel.from_pretrained(model_path, torch_dtype=torch.bfloat16, trust_remote_code=True)
63
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
64
  model = model.to("cuda").eval()