Adding pip install bitsandbytes for Colab and Kaggle compatibility

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