liwenyun
/
CLIP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

* 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 7 deletions
Show Stats 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 @@
] ]
} }
] ]
} }