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
- "# 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",
 
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
- openai
9
- peft
10
- ray
11
- shortuuid
12
- uvicorn
 
 
 
 
 
 
 
 
 
 
 
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