fix jax/flax

hollowstrawberry · Jan 8, 2025 · b4c4b85 · b4c4b85
1 parent 891808a
commit b4c4b85
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/Dataset_Maker.ipynb b/Dataset_Maker.ipynb
@@ -375,7 +375,7 @@
         "    print(\"\\n🏭 Installing dependencies...\\n\")\n",
         "    !pip install accelerate==0.25.0 diffusers[torch]==0.25.0 einops==0.6.0 tensorflow==2.15.0 \\\n",
         "      keras==2.15.0 transformers safetensors huggingface-hub==0.22.0 torchvision albumentations \\\n",
-        "      jax==0.4.23 jaxlib==0.4.23\n",
+        "      jax==0.4.23 jaxlib==0.4.23 flax==0.7.5\n",
         "    if not get_ipython().__dict__['user_ns']['_exit_code']:\n",
         "      clear_output()\n",
         "      step4a_installed_flag = True\n",
@@ -420,7 +420,7 @@
         "    print(\"\\n🏭 Installing dependencies...\\n\")\n",
         "    !pip install timm==0.6.12 fairscale==0.4.13 transformers==4.26.0 requests \\\n",
         "      accelerate==0.25.0 diffusers[torch]==0.25.0 einops==0.6.0 safetensors \\\n",
-        "      jax==0.4.23 jaxlib==0.4.23 huggingface-hub==0.22.0\n",
+        "      jax==0.4.23 jaxlib==0.4.23 flax==0.7.5 huggingface-hub==0.22.0\n",
         "    if not get_ipython().__dict__['user_ns']['_exit_code']:\n",
         "      clear_output()\n",
         "      step4b_installed_flag = True\n",

diff --git a/Lora_Trainer_XL.ipynb b/Lora_Trainer_XL.ipynb
@@ -289,7 +289,7 @@
         "  !apt -y update -qq\n",
         "  !apt -y install aria2 -qq\n",
         "  !pip install torch==2.5.1+cu121 accelerate==0.19.0 transformers==4.30.2 diffusers==0.18.2 \\\n",
-        "    bitsandbytes==0.40.0.post4 opencv-python jax==0.4.23 jaxlib==0.4.23 \\\n",
+        "    bitsandbytes==0.40.0.post4 opencv-python jax==0.4.23 jaxlib==0.4.23 flax==0.7.5 \\\n",
         "    pytorch-lightning==1.9.0 voluptuous==0.13.1 toml==0.10.2 ftfy==6.1.1 einops==0.6.0 \\\n",
         "    safetensors pygments huggingface-hub==0.22.0 wandb invisible-watermark==0.2.0 open-clip-torch==2.20.0 \\\n",
         "    dadaptation==3.1 prodigyopt==1.0 lion-pytorch==0.1.2\n",

diff --git a/Spanish_Dataset_Maker.ipynb b/Spanish_Dataset_Maker.ipynb
@@ -380,7 +380,7 @@
         "    print(\"🏭 Instalando...\\n\")\n",
         "    !pip install accelerate==0.25.0 diffusers[torch]==0.25.0 einops==0.6.0 tensorflow==2.15.0 \\\n",
         "      keras==2.15.0 transformers safetensors huggingface-hub==0.22.0 torchvision albumentations \\\n",
-        "      jax==0.4.23 jaxlib==0.4.23\n",
+        "      jax==0.4.23 jaxlib==0.4.23 flax==0.7.5\n",
         "    if not get_ipython().__dict__['user_ns']['_exit_code']:\n",
         "      clear_output()\n",
         "      step4a_installed_flag = True\n",
@@ -425,7 +425,7 @@
         "    print(\"🏭 Instalando...\\n\")\n",
         "    !pip install timm==0.6.12 fairscale==0.4.13 transformers==4.26.0 requests \\\n",
         "      accelerate==0.25.0 diffusers[torch]==0.25.0 einops==0.6.0 safetensors \\\n",
-        "      jax==0.4.23 jaxlib==0.4.23 huggingface-hub==0.22.0\n",
+        "      jax==0.4.23 jaxlib==0.4.23 flax==0.7.5 huggingface-hub==0.22.0\n",
         "    if not get_ipython().__dict__['user_ns']['_exit_code']:\n",
         "      clear_output()\n",
         "      step4b_installed_flag = True\n",
@@ -648,8 +648,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "7AdGcZRyPmtm",
-        "cellView": "form"
+        "cellView": "form",
+        "id": "7AdGcZRyPmtm"
       },
       "outputs": [],
       "source": [