dahara1 commited on
Commit
481f099
1 Parent(s): e46a468

Update finetune_sample/lora_test.py

Browse files
Files changed (1) hide show
  1. finetune_sample/lora_test.py +1 -1
finetune_sample/lora_test.py CHANGED
@@ -11,7 +11,7 @@ from transformers import GPTQConfig
11
  quantized_model_dir = "./"
12
  model_basename = "model"
13
 
14
- tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, , local_files_only=True)
15
 
16
  device_map = "auto"
17
  model = AutoModelForCausalLM.from_pretrained(
 
11
  quantized_model_dir = "./"
12
  model_basename = "model"
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, local_files_only=True)
15
 
16
  device_map = "auto"
17
  model = AutoModelForCausalLM.from_pretrained(