Update finetune_sample/lora_test.py
Browse files
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,
|
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(
|