capleaf
/

T-Llama

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

1TuanPham commited on Mar 20

Commit

5359d7e

•

1 Parent(s): c104028

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ Use the code below to get started with the model.
 from torch.cuda.amp import autocast
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer, pipeline
-model_name = "1TuanPham/InstructEnVi_llama2-bkai-120GB-Orcafamily_250kx3.37_350kx1.4"
 model = AutoModelForCausalLM.from_pretrained(model_name,
                                              torch_dtype=torch.bfloat16,
                                              use_cache=True,

 from torch.cuda.amp import autocast
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer, pipeline
+model_name = "1TuanPham/T-Llama"
 model = AutoModelForCausalLM.from_pretrained(model_name,
                                              torch_dtype=torch.bfloat16,
                                              use_cache=True,