Update README.md

README.md

@@ -12,7 +12,7 @@ license: llama3
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-BASE_MODEL = "./llama-3-korean-8b"
+BASE_MODEL = "sh2orc/llama-3-korean-8b"
 
 model = AutoModelForCausalLM.from_pretrained(BASE_MODEL,
     torch_dtype=torch.bfloat16,
@@ -72,7 +72,7 @@ print(outputs[0]['generated_text'][len(prompt):])
 from vllm import LLM, SamplingParams
 from transformers import AutoTokenizer, pipeline
 
-BASE_MODEL = "sh2orc/llama-3-korean-8b-awq"
+BASE_MODEL = "sh2orc/llama-3-korean-8b"
 
 llm = LLM(model=BASE_MODEL)