jingwang commited on
Commit
c55d3cb
·
verified ·
1 Parent(s): 7321327

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -0
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: