Commit
•
aa2e8ac
1
Parent(s):
be67426
Update README.md
Browse files
README.md
CHANGED
|
@@ -45,7 +45,7 @@ import pandas as pd
|
|
| 45 |
from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
| 46 |
from peft import LoraConfig, PeftModel
|
| 47 |
|
| 48 |
-
model_id = "harishnair04/Gemma-medtr-2b"
|
| 49 |
bnb_config = BitsAndBytesConfig(
|
| 50 |
load_in_4bit=True,
|
| 51 |
bnb_4bit_quant_type="nf4",
|
|
|
|
| 45 |
from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
|
| 46 |
from peft import LoraConfig, PeftModel
|
| 47 |
|
| 48 |
+
model_id = "harishnair04/Gemma-medtr-2b-sft"
|
| 49 |
bnb_config = BitsAndBytesConfig(
|
| 50 |
load_in_4bit=True,
|
| 51 |
bnb_4bit_quant_type="nf4",
|