Update README.md
Browse files
README.md
CHANGED
@@ -42,7 +42,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
42 |
base = AutoModelForCausalLM.from_pretrained("unsloth/gemma-2b-it", load_in_4bit=True, device_map="auto")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("unsloth/gemma-2b-it")
|
44 |
|
45 |
-
model = PeftModel.from_pretrained(base, "kihyun1998/gemma-2b-mongodb-lora")
|
46 |
|
47 |
prompt = """### Instruction:
|
48 |
Convert to MongoDB query string.
|
@@ -124,8 +124,8 @@ This LoRA adapter inherits the same conditions.
|
|
124 |
```bibtex
|
125 |
@misc{kihyun2025mongodb,
|
126 |
title={Gemma 2B MongoDB Query Generator (LoRA)},
|
127 |
-
author={Kihyun
|
128 |
year={2025},
|
129 |
-
howpublished={\\url{https://huggingface.co/kihyun1998/gemma-2b-mongodb-lora}}
|
130 |
}
|
131 |
```
|
|
|
42 |
base = AutoModelForCausalLM.from_pretrained("unsloth/gemma-2b-it", load_in_4bit=True, device_map="auto")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("unsloth/gemma-2b-it")
|
44 |
|
45 |
+
model = PeftModel.from_pretrained(base, "kihyun1998/gemma-2b-it-mongodb-lora")
|
46 |
|
47 |
prompt = """### Instruction:
|
48 |
Convert to MongoDB query string.
|
|
|
124 |
```bibtex
|
125 |
@misc{kihyun2025mongodb,
|
126 |
title={Gemma 2B MongoDB Query Generator (LoRA)},
|
127 |
+
author={Kihyun Park},
|
128 |
year={2025},
|
129 |
+
howpublished={\\url{https://huggingface.co/kihyun1998/gemma-2b-it-mongodb-lora}}
|
130 |
}
|
131 |
```
|