move huggingface import lower

hollowstrawberry · Nov 17, 2024 · 891808a · 891808a
1 parent 0751ef2
commit 891808a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Lora_Trainer_XL.ipynb b/Lora_Trainer_XL.ipynb
@@ -54,7 +54,6 @@
         "import toml\n",
         "import pathlib\n",
         "from time import time\n",
-        "from huggingface_hub import HfFileSystem\n",
         "from IPython.display import Markdown, display\n",
         "\n",
         "# These carry information from past executions\n",
@@ -549,6 +548,7 @@
         "              username = match.group(1)\n",
         "              model_name = match.group(2)\n",
         "              model_file = f\"{username}/{model_name}\"\n",
+        "              from huggingface_hub import HfFileSystem\n",
         "              fs = HfFileSystem()\n",
         "              existing_folders = set(fs.ls(model_file, detail=False))\n",
         "              necessary_folders = [ \"scheduler\", \"text_encoder\", \"text_encoder_2\", \"tokenizer\", \"tokenizer_2\", \"unet\", \"vae\" ]\n",