Spaces:
Running on T4
Running on T4
Fix recent colab errors
#11
by imgprcsng - opened
- notebooks/demo.ipynb +3 -3
- requirements.txt +2 -1
notebooks/demo.ipynb
CHANGED
|
@@ -133,7 +133,7 @@
|
|
| 133 |
"# TODO check for gcc-11 or above\n",
|
| 134 |
"\n",
|
| 135 |
"# Install pip packages\n",
|
| 136 |
-
"pip install --upgrade pip setuptools wheel\n",
|
| 137 |
"pip install -r requirements.txt\n",
|
| 138 |
"\n",
|
| 139 |
"cd models/GroundingDINO/ops\n",
|
|
@@ -143,12 +143,12 @@
|
|
| 143 |
" then\n",
|
| 144 |
" echo \"failed to install wheel, trying to build from source\";\n",
|
| 145 |
" python3 setup.py build\n",
|
| 146 |
-
"
|
| 147 |
" fi\n",
|
| 148 |
"else\n",
|
| 149 |
" # We try to build the module as we dont know what environment we are running on\n",
|
| 150 |
" python3 setup.py build\n",
|
| 151 |
-
"
|
| 152 |
"fi\n",
|
| 153 |
"python3 test.py"
|
| 154 |
]
|
|
|
|
| 133 |
"# TODO check for gcc-11 or above\n",
|
| 134 |
"\n",
|
| 135 |
"# Install pip packages\n",
|
| 136 |
+
"pip install --upgrade pip setuptools wheel ninja\n",
|
| 137 |
"pip install -r requirements.txt\n",
|
| 138 |
"\n",
|
| 139 |
"cd models/GroundingDINO/ops\n",
|
|
|
|
| 143 |
" then\n",
|
| 144 |
" echo \"failed to install wheel, trying to build from source\";\n",
|
| 145 |
" python3 setup.py build\n",
|
| 146 |
+
" python3 setup.py install\n",
|
| 147 |
" fi\n",
|
| 148 |
"else\n",
|
| 149 |
" # We try to build the module as we dont know what environment we are running on\n",
|
| 150 |
" python3 setup.py build\n",
|
| 151 |
+
" python3 setup.py install\n",
|
| 152 |
"fi\n",
|
| 153 |
"python3 test.py"
|
| 154 |
]
|
requirements.txt
CHANGED
|
@@ -17,5 +17,6 @@ tqdm
|
|
| 17 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 18 |
torch<2.6
|
| 19 |
torchvision
|
| 20 |
-
transformers
|
|
|
|
| 21 |
pydantic==2.10.6
|
|
|
|
| 17 |
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 18 |
torch<2.6
|
| 19 |
torchvision
|
| 20 |
+
transformers<5
|
| 21 |
+
huggingface_hub<1
|
| 22 |
pydantic==2.10.6
|