Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# Mistral Fine-Tuned Model
|
| 2 |
|
| 3 |
## Description
|
|
@@ -5,13 +16,10 @@ This is a fine-tuned version of Mistral 7B using Unsloth.
|
|
| 5 |
|
| 6 |
## Usage
|
| 7 |
|
|
|
|
| 8 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 9 |
|
| 10 |
model_id = "varundevmishra09/mistral-finetuned"
|
| 11 |
|
| 12 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 13 |
-
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 14 |
-
|
| 15 |
-
## Notes
|
| 16 |
-
- Fine-tuned using LoRA
|
| 17 |
-
- Merged into full model
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
base_model: mistralai/Mistral-7B-v0.1
|
| 4 |
+
tags:
|
| 5 |
+
- mistral
|
| 6 |
+
- fine-tuned
|
| 7 |
+
- unsloth
|
| 8 |
+
- text-generation
|
| 9 |
+
pipeline_tag: text-generation
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
# Mistral Fine-Tuned Model
|
| 13 |
|
| 14 |
## Description
|
|
|
|
| 16 |
|
| 17 |
## Usage
|
| 18 |
|
| 19 |
+
```python
|
| 20 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 21 |
|
| 22 |
model_id = "varundevmishra09/mistral-finetuned"
|
| 23 |
|
| 24 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 25 |
+
model = AutoModelForCausalLM.from_pretrained(model_id)
|
|
|
|
|
|
|
|
|
|
|
|