Update README.md
Browse files
README.md
CHANGED
|
@@ -52,11 +52,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained("LimYeri/CodeMind-Gemma-7B-QLoRA-4bit")
|
| 53 |
model = AutoModelForCausalLM.from_pretrained("LimYeri/CodeMind-Gemma-7B-QLoRA-4bit")
|
| 54 |
|
| 55 |
-
input_text = "Tell me how to solve the Leetcode Two Sum problem"
|
| 56 |
-
input_ids = tokenizer(input_text, return_tensors="pt")
|
| 57 |
-
|
| 58 |
-
outputs = model.generate(**input_ids)
|
| 59 |
-
print(tokenizer.decode(outputs[0]))
|
| 60 |
```
|
| 61 |
|
| 62 |
|
|
|
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained("LimYeri/CodeMind-Gemma-7B-QLoRA-4bit")
|
| 53 |
model = AutoModelForCausalLM.from_pretrained("LimYeri/CodeMind-Gemma-7B-QLoRA-4bit")
|
| 54 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
```
|
| 56 |
|
| 57 |
|