Update README.md
Browse files
README.md
CHANGED
@@ -27,7 +27,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
|
27 |
from peft import PeftModel
|
28 |
|
29 |
model_id = "google/gemma-1.1-7b-it"
|
30 |
-
peft_model_id = "brildev7/gemma-1.1-7b-it-translation-
|
31 |
quantization_config = BitsAndBytesConfig(
|
32 |
load_in_4bit=True,
|
33 |
bnb_4bit_compute_dtype=torch.float16,
|
|
|
27 |
from peft import PeftModel
|
28 |
|
29 |
model_id = "google/gemma-1.1-7b-it"
|
30 |
+
peft_model_id = "brildev7/gemma-1.1-7b-it-translation-koen-sft-qlora"
|
31 |
quantization_config = BitsAndBytesConfig(
|
32 |
load_in_4bit=True,
|
33 |
bnb_4bit_compute_dtype=torch.float16,
|