Update README.md
Browse files
README.md
CHANGED
|
@@ -49,7 +49,7 @@ Use the code below to get started with the model.
|
|
| 49 |
from torch.cuda.amp import autocast
|
| 50 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer, pipeline
|
| 51 |
|
| 52 |
-
model_name = "1TuanPham/InstructEnVi_llama2-bkai-120GB-Orcafamily_250kx3.37_350kx1.
|
| 53 |
model = AutoModelForCausalLM.from_pretrained(model_name,
|
| 54 |
torch_dtype=torch.bfloat16,
|
| 55 |
use_cache=True,
|
|
|
|
| 49 |
from torch.cuda.amp import autocast
|
| 50 |
from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer, pipeline
|
| 51 |
|
| 52 |
+
model_name = "1TuanPham/InstructEnVi_llama2-bkai-120GB-Orcafamily_250kx3.37_350kx1.4"
|
| 53 |
model = AutoModelForCausalLM.from_pretrained(model_name,
|
| 54 |
torch_dtype=torch.bfloat16,
|
| 55 |
use_cache=True,
|