Update README.md
Browse files
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/
|
| 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()
|