rename to ace step
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
import argparse
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--checkpoint_path", type=str, default="./checkpoints")
|
||||
parser.add_argument("--checkpoint_path", type=str, default="")
|
||||
parser.add_argument("--port", type=int, default=7865)
|
||||
parser.add_argument("--device_id", type=int, default=0)
|
||||
parser.add_argument("--share", action='store_true', default=False)
|
||||
@@ -14,13 +14,13 @@ os.environ["CUDA_VISIBLE_DEVICES"] = str(args.device_id)
|
||||
|
||||
|
||||
from ui.components import create_main_demo_ui
|
||||
from pipeline_fusic import FusicPipeline
|
||||
from pipeline_ace_step import ACEStepPipeline
|
||||
from data_sampler import DataSampler
|
||||
|
||||
|
||||
def main(args):
|
||||
|
||||
model_demo = FusicPipeline(
|
||||
model_demo = ACEStepPipeline(
|
||||
checkpoint_dir=args.checkpoint_path,
|
||||
dtype="bfloat16" if args.bf16 else "float32"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user