Update temporary installation instructions
Browse files- COLIPRI_demo.ipynb +10 -3
- README.md +2 -1
COLIPRI_demo.ipynb
CHANGED
|
@@ -32,7 +32,9 @@
|
|
| 32 |
},
|
| 33 |
"outputs": [],
|
| 34 |
"source": [
|
| 35 |
-
"
|
|
|
|
|
|
|
| 36 |
]
|
| 37 |
},
|
| 38 |
{
|
|
@@ -78,16 +80,21 @@
|
|
| 78 |
},
|
| 79 |
"outputs": [],
|
| 80 |
"source": [
|
|
|
|
|
|
|
| 81 |
"import torch\n",
|
| 82 |
"import torchio as tio\n",
|
| 83 |
"\n",
|
|
|
|
|
|
|
| 84 |
"from colipri import get_model\n",
|
| 85 |
"from colipri import get_processor\n",
|
| 86 |
"from colipri import load_sample_ct\n",
|
| 87 |
-
"from colipri import ZeroShotImageClassificationPipeline\n",
|
| 88 |
"\n",
|
| 89 |
"device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
|
| 90 |
-
"print(
|
|
|
|
|
|
|
| 91 |
]
|
| 92 |
},
|
| 93 |
{
|
|
|
|
| 32 |
},
|
| 33 |
"outputs": [],
|
| 34 |
"source": [
|
| 35 |
+
"# Temporary setup commands\n",
|
| 36 |
+
"!git clone https://huggingface.co/microsoft/colipri\n",
|
| 37 |
+
"%pip install --quiet ./colipri[demo]"
|
| 38 |
]
|
| 39 |
},
|
| 40 |
{
|
|
|
|
| 80 |
},
|
| 81 |
"outputs": [],
|
| 82 |
"source": [
|
| 83 |
+
"import time\n",
|
| 84 |
+
"\n",
|
| 85 |
"import torch\n",
|
| 86 |
"import torchio as tio\n",
|
| 87 |
"\n",
|
| 88 |
+
"import colipri\n",
|
| 89 |
+
"from colipri import ZeroShotImageClassificationPipeline\n",
|
| 90 |
"from colipri import get_model\n",
|
| 91 |
"from colipri import get_processor\n",
|
| 92 |
"from colipri import load_sample_ct\n",
|
|
|
|
| 93 |
"\n",
|
| 94 |
"device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
|
| 95 |
+
"print(\"Last run on\", time.ctime())\n",
|
| 96 |
+
"print(\"colipri version:\", colipri.__version__)\n",
|
| 97 |
+
"print(\"PyTorch device:\", device)"
|
| 98 |
]
|
| 99 |
},
|
| 100 |
{
|
README.md
CHANGED
|
@@ -52,7 +52,8 @@ Fine-tuning COLIPRI is typically not necessary to obtain good performance in dow
|
|
| 52 |
### Installation
|
| 53 |
|
| 54 |
```shell
|
| 55 |
-
|
|
|
|
| 56 |
```
|
| 57 |
|
| 58 |
### Usage examples
|
|
|
|
| 52 |
### Installation
|
| 53 |
|
| 54 |
```shell
|
| 55 |
+
git clone https://huggingface.co/microsoft/colipri
|
| 56 |
+
pip install --quiet ./colipri
|
| 57 |
```
|
| 58 |
|
| 59 |
### Usage examples
|