brildev7 commited on
Commit
d6e7b7c
1 Parent(s): a0bcf78

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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,