openai/CLIP
1
0
Fork 0
mirror of https://github.com/openai/CLIP.git synced 2026-04-29 20:01:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

use pkg_resources for PyTorch version checks in notebooks (#191)

Browse Source 
* use pkg_resources for PyTorch version checks in notebooks

* remove version checking in notebook code cell

Co-authored-by: Jong Wook Kim <jongwook@nyu.edu>
This commit is contained in:
neverix
2022-04-11 00:08:16 +03:00
committed by GitHub
parent 3482bb6ed3
commit e58d49454c
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
notebooks/Interacting_with_CLIP.ipynb
View File

@@ -352,10 +352,9 @@
"source": [ "source": [
"import numpy as np\n", "import numpy as np\n",
"import torch\n", "import torch\n",
"from pkg_resources import packaging\n",
"\n", "\n",
"print(\"Torch version:\", torch.__version__)\n", "print(\"Torch version:\", torch.__version__)\n"
"\n",
"assert torch.__version__.split(\".\") >= [\"1\", \"7\", \"1\"], \"PyTorch 1.7.1 or later is required\""
], ],
"execution_count": 2, "execution_count": 2,
"outputs": [ "outputs": [

7
notebooks/Prompt_Engineering_for_ImageNet.ipynb
View File

@@ -588,10 +588,9 @@
"import torch\n", "import torch\n",
"import clip\n", "import clip\n",
"from tqdm.notebook import tqdm\n", "from tqdm.notebook import tqdm\n",
"from pkg_resources import packaging\n",
"\n", "\n",
"print(\"Torch version:\", torch.__version__)\n", "print(\"Torch version:\", torch.__version__)\n"
"\n",
"assert torch.__version__.split(\".\") >= [\"1\", \"7\", \"1\"], \"PyTorch 1.7.1 or later is required\""
], ],
"execution_count": 2, "execution_count": 2,
"outputs": [ "outputs": [
@@ -1105,4 +1104,4 @@
] ]
} }
] ]
} }