ACE-Step/colab_inference.ipynb

Install

In [ ]:

#!pip uninstall ace-step -y
!pip install --upgrade git+https://github.com/craftmine1000/ACE-Step-Colab-Opt.git
import os
os.environ['COLAB_FREE'] = 'TRUE'

Import Model From GDrive (Optional)

In [ ]:

from google.colab import drive
drive.mount('/gdrive')
!unzip /gdrive/MyDrive/acestep/checkpoints.zip -d /unzip

Run Interface

In [ ]:

torch_compile = True # @param {type: "boolean"}
cpu_offload = True # @param {type: "boolean"}
overlapped_decode = True # @param {type: "boolean"}
#bf16 = True # @param {type: "boolean"}

!acestep --checkpoint_path /unzip/checkpoints/ --port 7865 --device_id 0 --share true --torch_compile {torch_compile} --cpu_offload {cpu_offload} --overlapped_decode {overlapped_decode}