harishnair04
commited on
Commit
•
be67426
1
Parent(s):
6bba74a
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 = "
|
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"
|
49 |
bnb_config = BitsAndBytesConfig(
|
50 |
load_in_4bit=True,
|
51 |
bnb_4bit_quant_type="nf4",
|