Fix recent colab errors

#11
by imgprcsng - opened
Files changed (2) hide show
  1. notebooks/demo.ipynb +3 -3
  2. 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
- " pip 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
- " pip install .\n",
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