Increase download timeout (#2585)

CVS-152443

---------

Co-authored-by: Andrei Kochin <[email protected]>

Author: aleksandr-mokrov

notebooks/blip-visual-language-processing/blip-visual-language-processing.ipynb

@@ -820,6 +820,8 @@
    "source": [
     "%%skip not $to_quantize.value\n",
     "\n",
+    "import aiohttp\n",
+    "\n",
     "import numpy as np\n",
     "from datasets import load_dataset\n",
     "from tqdm.notebook import tqdm\n",
@@ -881,7 +883,13 @@
     "    Prepares a vision-text dataset for quantization.\n",
     "    \"\"\"\n",
     "    split = f\"train[:{opt_init_steps}]\" if not streaming else \"train\"\n",
-    "    dataset = load_dataset(\"HuggingFaceM4/VQAv2\", split=split, streaming=streaming, trust_remote_code=True)\n",
+    "    dataset = load_dataset(\n",
+    "        \"HuggingFaceM4/VQAv2\", \n",
+    "        split=split, \n",
+    "        streaming=streaming, \n",
+    "        trust_remote_code=True,\n",
+    "        storage_options={'client_kwargs': {'timeout': aiohttp.ClientTimeout(total=3600)}}\n",
+    "    )\n",
     "    dataset = dataset.shuffle(seed=42)\n",
     "    if streaming:\n",
     "        dataset = dataset.take(opt_init_steps)\n",