Update inference.ipynb to fix bug and improve UI
There was a problem in the script that caused two lines to be merged. Fixed the problem. Also added some emojis to the cells
This commit is contained in:
+10
-5
@@ -81,8 +81,9 @@
|
|||||||
},
|
},
|
||||||
"outputs": [],
|
"outputs": [],
|
||||||
"source": [
|
"source": [
|
||||||
"#@title Install and Download\n",
|
"#@title 🎯 Install and Download\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
"from IPython.display import clear_output"
|
||||||
"\n",
|
"\n",
|
||||||
"import codecs\n",
|
"import codecs\n",
|
||||||
"\n",
|
"\n",
|
||||||
@@ -110,21 +111,25 @@
|
|||||||
"!uv pip install huggingface-hub numpy==1.26.0 > /dev/null 2>&1\n",
|
"!uv pip install huggingface-hub numpy==1.26.0 > /dev/null 2>&1\n",
|
||||||
"!huggingface-cli download ACE-Step/ACE-Step-v1-3.5B --local-dir /content/ACE-Step/checkpoints --local-dir-use-symlinks False\n",
|
"!huggingface-cli download ACE-Step/ACE-Step-v1-3.5B --local-dir /content/ACE-Step/checkpoints --local-dir-use-symlinks False\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
"!pip install e .",
|
||||||
|
"\n",
|
||||||
"\n",
|
"\n",
|
||||||
"import os\n",
|
"import os\n",
|
||||||
"os.environ['MPLBACKEND'] = 'agg'"
|
"os.environ['MPLBACKEND'] = 'agg'"
|
||||||
|
"\n\n"
|
||||||
|
"clear_output()"
|
||||||
|
"\n\n"
|
||||||
|
"print("✅ Installation Complete!")"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "code",
|
"cell_type": "code",
|
||||||
"source": [
|
"source": [
|
||||||
"#@title Run Gradio UI\n",
|
"#@title 🚀 Run Gradio UI\n",
|
||||||
"bf16 = True # @param {\"type\":\"boolean\"}\n",
|
"bf16 = True # @param {\"type\":\"boolean\"}\n",
|
||||||
"\n",
|
"\n",
|
||||||
|
"print(\" 🚀 Running UI...\")\n",
|
||||||
"\n",
|
"\n",
|
||||||
"print(\" * Running UI...\")\n",
|
|
||||||
"\n",
|
|
||||||
"!pip install e .",
|
|
||||||
"!acestep --checkpoint_path ./checkpoints/ --port 7865 --device_id 0 --share true --bf16 {bf16}"
|
"!acestep --checkpoint_path ./checkpoints/ --port 7865 --device_id 0 --share true --bf16 {bf16}"
|
||||||
],
|
],
|
||||||
"metadata": {
|
"metadata": {
|
||||||
|
|||||||
Reference in New Issue
Block a user