kihyun1998 commited on
Commit
02012e9
·
verified ·
1 Parent(s): cd62380

Update README.md

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