Update README.md
Browse files
README.md
CHANGED
|
@@ -42,7 +42,7 @@ from peft import PeftModel
|
|
| 42 |
|
| 43 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-7B-Instruct")
|
| 44 |
model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-7B-Instruct",
|
| 45 |
-
torch_dtype="auto", device_map="auto",
|
| 46 |
|
| 47 |
model = PeftModel.from_pretrained(model, "svjack/Qwen2-7B_Function_Call_tiny_lora")
|
| 48 |
streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
|
|
|
|
| 42 |
|
| 43 |
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-7B-Instruct")
|
| 44 |
model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2-7B-Instruct",
|
| 45 |
+
torch_dtype="auto", device_map="auto", load_in_8bit = True)
|
| 46 |
|
| 47 |
model = PeftModel.from_pretrained(model, "svjack/Qwen2-7B_Function_Call_tiny_lora")
|
| 48 |
streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
|