partex-nv
/

ClinicalGPT-Pubmed-Instruct-V1.0

Question Answering

Model card Files Files and versions Community

partex-nv commited on 23 days ago

Commit

b0ab654

•

1 Parent(s): c6f7316

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 # Set parameters
-model_dir = "rohitanurag/ClinicalGPT-Pubmed-Instruct-V1.0"
 max_new_tokens = 1500
 device = "cuda" if torch.cuda.is_available() else "cpu"

 import torch
 # Set parameters
+model_dir = "partex-nv/ClinicalGPT-Pubmed-Instruct-V1.0"
 max_new_tokens = 1500
 device = "cuda" if torch.cuda.is_available() else "cpu"