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",
|