diff --git a/demo.ipynb b/demo.ipynb index b230bb8..80244ef 100644 --- a/demo.ipynb +++ b/demo.ipynb @@ -192,7 +192,7 @@ { "cell_type": "code", "source": [ - "!python app.py --device \"cuda\" --max-gen 4096 --share" + "!python app.py --max-gen 4096 --share" ], "metadata": { "id": "XUSb-v_cXi4k",