diff --git a/scripts/rubik_conversion.ipynb b/scripts/rubik_conversion.ipynb index f1fef0fe4..ce954042c 100644 --- a/scripts/rubik_conversion.ipynb +++ b/scripts/rubik_conversion.ipynb @@ -27,8 +27,6 @@ "\n", "**NOTE: your API key will be listed in the output, and should therefore be redacted if the output is shared.**\n", "\n", - "At some point during the run of this cell, it will prompt you to choose whether to clone a repo. You should click on the text box next to the prompt, and answer *y/yes*.\n", - "\n", "Once the run has finished, open the AI Hub link, and download the tflite model for the job you just ran.\n", "\n", "If you want to use this notebook to convert a yolo11 model, you'll need to replace all instances of `yolov8` in the cell below with `yolov11`." @@ -52,7 +50,7 @@ "# sets up AI Hub enviroment\n", "!qai-hub configure --api_token \n", "# Converts the model to be ran on RB3Gen2\n", - "!python -m qai_hub_models.models.yolov8_det.export --quantize w8a8 --device=\"RB3 Gen 2 (Proxy)\" --ckpt-name /PATH/TO/WEIGHTS --device-os linux --target-runtime tflite --output-dir .\n" + "!yes | python -m qai_hub_models.models.yolov8_det.export --quantize w8a8 --device=\"RB3 Gen 2 (Proxy)\" --ckpt-name /PATH/TO/WEIGHTS --device-os linux --target-runtime tflite --output-dir .\n" ] }, {