Adding pip install bitsandbytes for Colab and Kaggle compatibility
#10
by
spkshumway
- opened
- 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",
|