Update README.md
Browse files
README.md
CHANGED
@@ -52,7 +52,7 @@ Today Date: 07 Feb 2025
|
|
52 |
```
|
53 |
import torch
|
54 |
from peft import PeftModel
|
55 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
56 |
|
57 |
quantization_config = BitsAndBytesConfig(
|
58 |
load_in_4bit=True,
|
|
|
52 |
```
|
53 |
import torch
|
54 |
from peft import PeftModel
|
55 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
|
56 |
|
57 |
quantization_config = BitsAndBytesConfig(
|
58 |
load_in_4bit=True,
|