Spaces:
Paused
Paused
alessandro trinca tornidor
commited on
Commit
·
c5dcce1
1
Parent(s):
2d83d09
[test] rename old notebook to tests_saturncloud.ipynb, add tests_colab.ipynb, update requirements_colab.txt
Browse files
notebooks/tests_colab.ipynb
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
notebooks/{test.ipynb → tests_saturncloud.ipynb}
RENAMED
|
@@ -559,7 +559,7 @@
|
|
| 559 |
"print(f\"start inference using inference_fn, now is {datetime.now()}.\")\n",
|
| 560 |
"idx_example = 0\n",
|
| 561 |
"input_prompt, input_image_path = constants.examples[idx_example]\n",
|
| 562 |
-
"#
|
| 563 |
"print(f\"input_prompt:{input_prompt}.\")\n",
|
| 564 |
"output_image, output_mask, output_str = inference_fn(\n",
|
| 565 |
" input_prompt,\n",
|
|
|
|
| 559 |
"print(f\"start inference using inference_fn, now is {datetime.now()}.\")\n",
|
| 560 |
"idx_example = 0\n",
|
| 561 |
"input_prompt, input_image_path = constants.examples[idx_example]\n",
|
| 562 |
+
"# input_prompt = \"Assuming you are an autonomous driving robot, what part of the diagram would you manipulate to control the direction of travel? Please output segmentation mask and explain why.\"\n",
|
| 563 |
"print(f\"input_prompt:{input_prompt}.\")\n",
|
| 564 |
"output_image, output_mask, output_str = inference_fn(\n",
|
| 565 |
" input_prompt,\n",
|
requirements_colab.txt
CHANGED
|
@@ -1,12 +1,22 @@
|
|
| 1 |
-
bitsandbytes
|
| 2 |
-
einops
|
| 3 |
-
fastapi
|
| 4 |
-
gradio
|
| 5 |
-
gradio_client
|
| 6 |
-
markdown2
|
| 7 |
-
nh3
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
bitsandbytes==0.42.0
|
| 2 |
+
einops==0.7.0
|
| 3 |
+
fastapi==0.109.2
|
| 4 |
+
gradio==4.19.1
|
| 5 |
+
gradio_client==0.10.0
|
| 6 |
+
markdown2==2.4.12
|
| 7 |
+
nh3==0.2.15
|
| 8 |
+
numpy==1.24.4
|
| 9 |
+
openai==1.12.0
|
| 10 |
+
opencv-python==4.9.0.80
|
| 11 |
+
packaging
|
| 12 |
+
peft==0.8.2
|
| 13 |
+
pycocotools==2.0.7
|
| 14 |
+
ray==2.9.2
|
| 15 |
+
scipy==1.12.0
|
| 16 |
+
sentencepiece
|
| 17 |
+
shortuuid==1.0.11
|
| 18 |
+
torch==2.2.0
|
| 19 |
+
torchvision==0.17.0
|
| 20 |
+
tqdm==4.66.2
|
| 21 |
+
transformers==4.31.0
|
| 22 |
+
uvicorn==0.27.1
|