Other
English
minecraft
action prediction
Kqte commited on
Commit
7ac2d98
·
verified ·
1 Parent(s): 06d8b1e

Update parser_generate.py

Browse files

fixed line 9 to show path to llamipa adapter

Files changed (1) hide show
  1. 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/meta-llama3-8b/",
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)