Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ from datasets import load_dataset
|
|
35 |
from peft import LoraConfig
|
36 |
from trl import SFTTrainer
|
37 |
|
38 |
-
model_id = "brildev7/gemma-7b-it-finetune-summarization-ko"
|
39 |
quantization_config = BitsAndBytesConfig(
|
40 |
load_in_4bit=True,
|
41 |
bnb_4bit_compute_dtype=torch.float16,
|
|
|
35 |
from peft import LoraConfig
|
36 |
from trl import SFTTrainer
|
37 |
|
38 |
+
model_id = "brildev7/gemma-7b-it-finetune-summarization-ko-lora"
|
39 |
quantization_config = BitsAndBytesConfig(
|
40 |
load_in_4bit=True,
|
41 |
bnb_4bit_compute_dtype=torch.float16,
|