MonteXiaofeng commited on
Commit
1749d90
1 Parent(s): a240c49

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -64,7 +64,7 @@ llama3_jinja = """{% if messages[0]['role'] == 'system' %}
64
 
65
  dtype = torch.bfloat16
66
 
67
- model_dir = "MonteXiaofeng/Hospitality-llama3_1_8B_instruct"
68
  model = AutoModelForCausalLM.from_pretrained(
69
  model_dir,
70
  device_map="cuda",
 
64
 
65
  dtype = torch.bfloat16
66
 
67
+ model_dir = "MonteXiaofeng/Law_Justice-llama3_1_8B_instruct"
68
  model = AutoModelForCausalLM.from_pretrained(
69
  model_dir,
70
  device_map="cuda",