Adding pip install bitsandbytes for Colab and Kaggle compatibility (#10)
Browse files- Adding pip install bitsandbytes for Colab and Kaggle compatibility (dc813be5bd74fc11d2e7e178741e84ca34026f17)
Co-authored-by: Spencer Shumway <spkshumway@users.noreply.huggingface.co>
- notebook.ipynb +1 -0
notebook.ipynb
CHANGED
|
@@ -112,6 +112,7 @@
|
|
| 112 |
}
|
| 113 |
],
|
| 114 |
"source": [
|
|
|
|
| 115 |
"from transformers import AutoModelForCausalLM, AutoTokenizer\n",
|
| 116 |
"\n",
|
| 117 |
"MODEL_NAME = 'NousResearch/Genstruct-7B'\n",
|
|
|
|
| 112 |
}
|
| 113 |
],
|
| 114 |
"source": [
|
| 115 |
+
"!pip install -U bitsandbytes --no-deps\n",
|
| 116 |
"from transformers import AutoModelForCausalLM, AutoTokenizer\n",
|
| 117 |
"\n",
|
| 118 |
"MODEL_NAME = 'NousResearch/Genstruct-7B'\n",
|