Update parser_generate.py
Browse filesfixed line 9 to show path to llamipa adapter
- parser_generate.py +1 -1
parser_generate.py
CHANGED
@@ -6,7 +6,7 @@ from tqdm import tqdm
|
|
6 |
|
7 |
device_map = "auto"
|
8 |
model = AutoModelForCausalLM.from_pretrained(
|
9 |
-
"/path/to/
|
10 |
return_dict=True,
|
11 |
torch_dtype=torch.float16,
|
12 |
device_map=device_map)
|
|
|
6 |
|
7 |
device_map = "auto"
|
8 |
model = AutoModelForCausalLM.from_pretrained(
|
9 |
+
"/path/to/llamipa/adapter",
|
10 |
return_dict=True,
|
11 |
torch_dtype=torch.float16,
|
12 |
device_map=device_map)
|