jinyuan22
/

RFamLlama-base

Text Generation

text-generation-inference

Model card Files Files and versions

jinyuan22 commited on May 19, 2024

Commit

923e19e

·

verified ·

1 Parent(s): d2cd241

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ from transformers import LlamaForCausalLM, AutoTokenizer, pipeline
 import torch
 import sys
-model_url = ""
 model = LlamaForCausalLM.from_pretrained(model_url, torch_dtype=torch.float16)
 tokenizer = AutoTokenizer.from_pretrained(model_url)

 import torch
 import sys
+model_url = "jinyuan22/RFamLlama-base"
 model = LlamaForCausalLM.from_pretrained(model_url, torch_dtype=torch.float16)
 tokenizer = AutoTokenizer.from_pretrained(model_url)