EmergenceAI
/

RAG-response-generation-model-v1

Text Generation

text-generation-inference

Model card Files Files and versions Community

aditya-emergence commited on Jun 23

Commit

a825b63

•

1 Parent(s): 4e5dbb0

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ Loading model and tokenizer:
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-model_path = "MerlynMind/Merlyn-QA_from_Corpus_Bio_History-llama2-13b"
 device = torch.device("cuda:0") # change device id as necessary
 model = AutoModelForCausalLM.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path, fast_tokenizer=True)

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+model_path = "EmergenceAI/RAG-response-generation-model-v1"
 device = torch.device("cuda:0") # change device id as necessary
 model = AutoModelForCausalLM.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path, fast_tokenizer=True)