MonteXiaofeng
commited on
Commit
•
1749d90
1
Parent(s):
a240c49
Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ llama3_jinja = """{% if messages[0]['role'] == 'system' %}
|
|
64 |
|
65 |
dtype = torch.bfloat16
|
66 |
|
67 |
-
model_dir = "MonteXiaofeng/
|
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",
|