teknium spkshumway commited on
Commit
cd4c3d7
·
verified ·
1 Parent(s): 532dc45

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>

Files changed (1) hide show
  1. 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",