diff --git a/Train_Colab.ipynb b/Train_Colab.ipynb index 2b83f6f..9c2cae5 100644 --- a/Train_Colab.ipynb +++ b/Train_Colab.ipynb @@ -107,9 +107,9 @@ "\n", "packages = [\n", " 'transformers==4.29.2',\n", - " 'diffusers[torch]==0.14.0',\n", + " 'diffusers[torch]==0.17.1',\n", " 'pynvml==11.4.1',\n", - " 'bitsandbytes==0.37.2',\n", + " 'bitsandbytes==0.38.1',\n", " 'ftfy==6.1.1',\n", " 'aiohttp==3.8.4',\n", " 'compel~=1.1.3',\n",