Update README.md
Browse files
README.md
CHANGED
|
@@ -19,8 +19,17 @@ base_model: unsloth/mistral-7b-v0.3-bnb-4bit
|
|
| 19 |
|
| 20 |
This mistral model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
|
| 21 |
|
|
|
|
|
|
|
|
|
|
| 22 |
```python
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
class FormatPrompt_context_QA():
|
| 25 |
'''format prompt class'''
|
| 26 |
def __init__(self, eos_token:str='</s>') -> None:
|
|
|
|
| 19 |
|
| 20 |
This mistral model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
|
| 21 |
|
| 22 |
+
$!pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
|
| 23 |
+
$!pip install --no-deps xformers "trl<0.9.0" peft accelerate bitsandbytes
|
| 24 |
+
|
| 25 |
```python
|
| 26 |
|
| 27 |
+
from unsloth import FastLanguageModel
|
| 28 |
+
from typing import Dict, List, Tuple, Union, Any
|
| 29 |
+
import pandas
|
| 30 |
+
from tqdm import trange, tqdm
|
| 31 |
+
import torch
|
| 32 |
+
|
| 33 |
class FormatPrompt_context_QA():
|
| 34 |
'''format prompt class'''
|
| 35 |
def __init__(self, eos_token:str='</s>') -> None:
|